ImitationEvaluator

class maze.train.trainers.imitation.imitation_evaluator.ImitationEvaluator

Abstract interface for imitation learning evaluation.

evaluate(policy: maze.core.agent.torch_policy.TorchPolicy) → None

Evaluate given policy (results are stored in stat logs) and dump the model if the reward improved.

Parameters

policy – Policy to evaluate