Skip to content
This repository was archived by the owner on Sep 20, 2023. It is now read-only.

liukanshan1/Fed-Avg

Repository files navigation

联邦学习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

About

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

Resources

Stars

21 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages