File size: 840 Bytes
041508c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
# model configs
from configurations.fastai_configs import conf_fastai_inception1d
from experiments.scp_experiment import SCPExperiment
from utilities.utils import generate_ptbxl_summary_table
def main():
data_folder = 'data/ptbxl/'
output_folder = 'output/'
models = [conf_fastai_inception1d]
# STANDARD SCP EXPERIMENTS ON PTB-XL
experiments = [
('exp0', 'all'),
('exp1', 'diagnostic'),
('exp1.1', 'subdiagnostic'),
('exp1.1.1', 'superdiagnostic'),
('exp2', 'form'),
('exp3', 'rhythm')
]
for name, task in experiments:
e = SCPExperiment(name, task, data_folder, output_folder, models)
e.prepare()
e.perform()
e.evaluate()
# generate summary table
generate_ptbxl_summary_table()
if __name__ == "__main__":
main() |