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)