Xiaotian-Ma / ECG-1

ECG classification

ECG classification

• 心电图进行疾病预测分类，共分为四类：AF，Normal, Noise, Others.
• 使用DNN网络，实现可以直接输出预测疾病名称与其可信度。
• 此版代码暂未经优化。

Environment

python 2.7

Install

Clone the repository

git clone [e-mail]:QiujieDong/ECG.git

Install the requirements (this may take a few minutes).

For CPU only support run

./setup.sh

To install with GPU support run

env TF=gpu ./setup.sh

Training

In the repo root direcotry (ecg) make a new directory called saved.

mkdir saved

To train a model use the following command, replacing path_to_config.json with an actual config:

python ecg/train.py path_to_config.json

Note that after each epoch the model is saved in ecg/saved/<experiment_id>/<timestamp>/<model_id>.hdf5.

For an actual example of how to run this code on a real dataset, you can follow the instructions in the cinc17 README. This will walk through downloading the Physionet 2017 challenge dataset and training and evaluating a model.

Testing

After training the model for a few epochs, you can make predictions with.

python ecg/predict.py <dataset>.json <model>.hdf5

replacing <dataset> with an actual path to the dataset and <model> with the path to the model.

参考：

[StanfordML Group]