def diagnose_model(self, horizon):
"""
Play a game only with the learned model then play the same trajectory in the real
environment and display information.
Args:
horizon (int): Number of timesteps for which we collect information.
"""
game = self.Game(self.config.seed)
obs = game.reset()
dm = diagnose_model.DiagnoseModel(self.checkpoint, self.config)
dm.compare_virtual_with_real_trajectories(obs, game, horizon)
input("Press enter to close all plots")
dm.close_all()
def diagnose_model(self, horizon):
"""
Play a game only with the learned model then play the same trajectory in the real
environment and display information.
Args:
horizon: number of timesteps for which we collect information.
"""
game = self.Game(self.config.seed)
obs = game.reset()
diagnose_model.DiagnoseModel(
self.muzero_weights,
self.config).compare_virtual_with_real_trajectories(
obs, game, horizon)
