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