This folder contain the deep neural network predictions on the test set. All files are in
the format .npy
and can be read using numpy.load()
. Each one should contain a
All the content within this folder can be generate using the following sequence of commands:
(without a GPU it should take about 25 minutes. With GPU acceleration it should take
less then one minute)
```bash
cd /path/to/automatic-ecg-diagnosis
PFOLDER="./dnn_predicts"
MFOLDER="./model"
DFOLDER="./data"
python predict.py $DFOLDER/ecg_tracings.hdf5 $MFOLDER/model.hdf5 --output_file $PFOLDER/model.npy
mkdir $FNAME/other_seeds
for n in 1 2 3 4 5 6 7 8 9 10
do
python predict.py $DFOLDER/ecg_tracings.hdf5 $MFOLDER/other_seeds/model_$n.hdf5 --output_file $PFOLDER/other_seeds/model_$n.npy
done
mkdir $PFOLDER/other_splits
for n in date_order individual_patients normal_order
do
python predict.py $DFOLDER/ecg_tracings.hdf5 $MFOLDER/other_splits/model_$n.hdf5 --output_file $PFOLDER/other_splits/model_$n.npy
done
```
Where the DFOLDER
should give the path to the folder containing the test dataset and MFOLDER
should point to the
folder containing pre-trained models. The test dataset can be downloaded from here and the
pretrained models can be downloaded from here here