deepr.exporters.BestCheckpoint

class deepr.exporters.BestCheckpoint(metric, mode=BestMode.DECREASE, use_mlflow=False, tag='training_best_step')[source]

Overrides Checkpoint Information to point to the best checkpoint.

The best checkpoint is the one whose step is closest to the step of the best evaluation metric. The evaluation metrics are retrieved from the evaluation summaries (located in eval).

metric

Name of the metric used to select the best checkpoint

Type:: str

mode

If ‘decrease’, means lower is better.

Type:: str

tag

Tag for MLFlow

Type:: str

use_mlflow

If True, will log the best checkpoint step as tag to MLFlow

Type:: bool

__init__(metric, mode=BestMode.DECREASE, use_mlflow=False, tag='training_best_step')[source]

Methods

`__init__`(metric[, mode, use_mlflow, tag])
`export`(estimator)