search

mondejar / ecg-classification

Code for training and test machine learning classifiers on MIT-BIH Arrhyhtmia database
GNU General Public License v3.0
520 stars 193 forks source link

No data for 'Q' Class #7

Closed banivyom closed 5 years ago

banivyom commented 5 years ago

When I delete all pickle files and then edit the load_MITBIH.py to include the fifth(Q) class, only 4 classes get included and not the fifth. I have also updated the evaluation_AAMI.py for confusion matrix for 5 class, but fifth row and column contains 0 always. I'm taking window size of 180 centred at R-peak.

What should I do?

mondejar commented 5 years ago

Check carefully if there are instances labelled with Q class in each dataset. Due to the window size and the database division, it's possible that no Q samples are being used during training.