wormpose.machine_learning.best_models_saver

Implements a Keras callback to save the top N best models on evaluation data,

class wormpose.machine_learning.best_models_saver.BestModels(models_dir, models_to_keep=5)[source]
on_epoch_end(epoch, logs=None)[source]

At the end of each epoch, update the best models and save to json