ESP3201_Project

This is a project from a module called ESP3902 which basically be used to do the ECG classfication for Arrhythmia using CNN

ESP3201 NUS Group 9

Haolin Chen, Mejsnarova Valerie

for Machine Learning Purpose and testing as well.

everything will appear, just relax and try to fix it

Meaning of each dataset

Small_dataset:

S_data_labels_cb_bw means with label combined train and test together and in black and write

S_prediction_datas_labels means with label for prediction

S_prediction_datas_labels_bw means with label and in black and write for prediction

S_testing_datas_labels means with label not combined train and test together

S_training_datas_labels means with label not combined train and test together

Model

model_20E_fulldata.h5 means model trained with 20 epochs and full dataset

model_test_smalldata.h5 means model trained with 20 epochs and small dataset

model_20E_6_4_bw.h5 means model trained with 20 epochs and 60% for training 40% for testing(black and white)

model_test_smalldata_bw.h5 means model trained with 20 epochs and small bw dataset, 80% for training 20% for testing

model_50E_8_2_bw.h5 means model trained with 50 epochs and 80% for training 20% for testing(black and white)

model_new_20E_8_2_bw.h5 means model trained with 20 epochs and 80% for training 20% for testing(black and white) with high accuracy!!(1st)

model_10E_8_2_bw_augmented means model trained with 10 epochs and 80% for training 20% for testing(black and white) with augmentationed 5000 dataset(2nd)

model_20E_8_2_bw_augmented means model trained with 20 epochs and 80% for training 20% for testing(black and white) with augmentationed 5000 dataset(3rd)

model_new_20E_803dataset means model trained with 20 epochs and 80% for training 20% for testing(black and white) with augmentationed 803 dataset(4th)

model_NN_lastone.h5 meansmodel trained with one more convolution layers, 20 epochs and 80% for training 20% for testing(black and white) with augmentationed 5000 dataset(5th)