updates to the ML Model

uhh-cms / hh2bbww

HH -> bbWW Analysis based on columnflow, law and order.

GNU General Public License v3.0

6 stars 9 forks source link

Closed mafrahm closed 1 year ago

mafrahm commented 1 year ago

This PR adds:

typehints for MLModel parameters
steps_per_epoch parameter
Mixin to allow training with standard tf dataset (TODO: this also requires to change the ml weights)
improved implementation of Keras Callbacks
allow using the "best model" from Checkpoint Callback