diff --git a/ANN_Training.py b/ANN_Training.py index c7961fdbcd84712312fa0ea5f1f0fb264a2ddc98..8a7ba02b30e16e89ef3a9f91fc4712bca117e6b5 100644 --- a/ANN_Training.py +++ b/ANN_Training.py @@ -2,19 +2,13 @@ """ @author: Laura C. Kühle, Soraya Terrab (sorayaterrab) -TODO: Add ANN classification from Soraya TODO: Give option to compare multiple models TODO: Use sklearn for classification TODO: Fix difference between accuracies (stems from rounding; choose higher value instead) TODO: Add more evaluation measures (AUROC, ROC, F1, training accuracy, etc.) -TODO: Decide on k-fold cross-validation (Use? Which model do we keep?) -> Done (CV only for testing) -TODO: Rework model testing to use k-fold cross-validation -> Done -TODO: Clean up directories/naming -> Done TODO: Add log to pipeline TODO: Remove object set-up -TODO: Add testing rule to Snakefile -> Done -TODO: Give option to change 'num_epochs' for testing -> Done -TODO: Adapt plot labels to new set-up -> Done +TODO: Optimize Snakefile-vs-config relation """ import numpy as np