BestModelSelection

class maze.train.trainers.common.model_selection.best_model_selection.BestModelSelection(dump_file: Optional[str], model: Optional[maze.core.agent.torch_model.TorchModel])

Best model selection strategy.

Parameters

• dump_file – Specifies the file path to dump the policy state for the best reward.

• model – The model to be dumped.

update(reward: float) → None

Implementation of ModelSelection.update().