antonior92 / ecg-age-prediction

Scripts and modules for training and testing neural network for age prediction from the ECG. Companion code to the paper "Deep neural network-estimated electrocardiographic age as a mortality predictor".
https://www.nature.com/articles/s41467-021-25351-7
MIT License
60 stars 23 forks source link

train.py does not track MAE for later use in plot_learning_curves.py #4

Open elijahrockers opened 9 months ago

elijahrockers commented 9 months ago

Getting an error for plot_learning_curves.py because train.py does not seem to append MAE to the dataframe. I am not able to tell where MAE is being calculated.