From c0099d2ff380840650a6d29e95b5f955c97bb4c9 Mon Sep 17 00:00:00 2001 From: SP Mohanty <spmohanty91@gmail.com> Date: Tue, 30 Jul 2019 05:06:15 +0200 Subject: [PATCH] Round off scores --- flatland/evaluators/service.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py index 21a5e63f..e4cab94c 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 -- GitLab