Closed HardiVv closed 1 month ago
The task was the same as in #62 but for binary classification.
(JP: if prediction values are very small/large in the initial epochs, tensorboard histograms are not able to visualize it correctly)
(JP: train being significantly worse than validation seems like a bug)
Train ROC curves
Validation ROC curves. Kinematics + features is better at low background, while Kinematics + features + lifetimes performs better overall. Choosing the right feature set is crucial depending on the model's objective.
Ran 3 binary classification tasks on three feature sets
Each run took about a day and 150 GB memory