fusionlab.datasets.cinc2017 module#

class fusionlab.datasets.cinc2017.ECGCSVClassificationDataset(data_root, label_filename='REFERENCE-v3.csv', channels=['lead'], class_names=['N', 'O', 'A', '~'])[source]#

Bases: Dataset

__init__(data_root, label_filename='REFERENCE-v3.csv', channels=['lead'], class_names=['N', 'O', 'A', '~'])[source]#
Parameters:

  • data_root (str) – root directory of the dataset

  • label_filename (str) – filename of the label file

  • channels (list) – list of target lead names

  • class_names (list) – list of class names for mapping class name to class id

fusionlab.datasets.cinc2017.convert_mat_to_csv(root, target_dir='csv')[source]#
fusionlab.datasets.cinc2017.validate_data(csv_dir, label_path)[source]#

check if the number of csv files and label files are matched