EvaluateModel module

EvaluateModel.evaluate_model(model, env, num_episodes=1)

Evaluate a trained model on a specified environment.

Parameters

modelstable_baselines3.common.base_class.BaseAlgorithm

The trained model to be evaluated.

envgym.Env

The Gym environment on which the model is evaluated.

num_episodesint, optional

The number of episodes for which the model is evaluated, by default 1.

Returns

float

The mean reward achieved by the model over all episodes.

list of list

The list of actions taken by the model in each episode.