ksuit.callbacks.early_stoppers.metric_early_stopper

Classes

MetricEarlyStopper

Early stopper (training) based on a metric value to be monitored.

Module Contents

class ksuit.callbacks.early_stoppers.metric_early_stopper.MetricEarlyStopper(callback_config, **kwargs)

Bases: ksuit.callbacks.early_stoppers.early_stopper_base.EarlyStopperBase

Early stopper (training) based on a metric value to be monitored.

Initialize the early stopper.

Parameters:

  • metric_key – The key of the metric to monitor.

  • tolerance – The number of times the metric can stagnate before stopping training.

  • metric_property_provider – The metric property provider to use.

  • **kwargs – Additional arguments to pass to the parent class.

  • callback_config (ksuit.schemas.callbacks.callbacks_config.MetricEarlyStopperConfig)

metric_key
higher_is_better
tolerance
tolerance_counter = 0
best_metric