diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py index 21a5e63f3a0e58ff915c3743eac56250435b5ee4..e4cab94c6915e560efde54dbfe8944a8ea2ef3d2 100644 --- a/flatland/evaluators/service.py +++ b/flatland/evaluators/service.py @@ -352,9 +352,9 @@ class FlatlandRemoteEvaluationService: progress = np.clip( self.simulation_count * 1.0 / len(self.env_file_paths), 0, 1) - mean_reward = np.mean(self.simulation_rewards) - mean_normalized_reward = np.mean(self.simulation_rewards_normalized) - mean_percentage_complete = np.mean(self.simulation_percentage_complete) + mean_reward = round(np.mean(self.simulation_rewards), 2) + mean_normalized_reward = round(np.mean(self.simulation_rewards_normalized), 2) + mean_percentage_complete = round(np.mean(self.simulation_percentage_complete), 3) self.evaluation_state["state"] = "IN_PROGRESS" self.evaluation_state["progress"] = progress self.evaluation_state["simulation_count"] = self.simulation_count @@ -457,9 +457,9 @@ class FlatlandRemoteEvaluationService: """ ) - mean_reward = np.mean(self.simulation_rewards) - mean_normalized_reward = np.mean(self.simulation_rewards_normalized) - mean_percentage_complete = np.mean(self.simulation_percentage_complete) + mean_reward = round(np.mean(self.simulation_rewards), 2) + mean_normalized_reward = round(np.mean(self.simulation_rewards_normalized), 2) + mean_percentage_complete = round(np.mean(self.simulation_percentage_complete), 3) if self.visualize and len(os.listdir(self.vizualization_folder_name)) > 0: # Generate the video