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