daojixing / Fed-Avg

联邦学习ECG心电图诊断模型训练与预测

联邦学习ECG心电图诊断模型训练与预测

采用明文模型平均，模型结构和部分参考论文Automatic diagnosis of the 12-lead ECG using a deep neural network及其开源实现

Scripts

• train.py: Script for training the neural network. To train the neural network run:

$ python train.py .\data\ .\data\

• predict.py: Script for generating the neural network predictions on a given dataset.

$ python predict.py .\data\test_set\ .\final_model.hdf5

• generate_figures_and_tables.py: Generate results.

$ python generate_results.py