Skip to content
Snippets Groups Projects
Commit c0099d2f authored by spmohanty's avatar spmohanty
Browse files

Round off scores

parent 96feebff
No related branches found
No related tags found
No related merge requests found
...@@ -352,9 +352,9 @@ class FlatlandRemoteEvaluationService: ...@@ -352,9 +352,9 @@ class FlatlandRemoteEvaluationService:
progress = np.clip( progress = np.clip(
self.simulation_count * 1.0 / len(self.env_file_paths), self.simulation_count * 1.0 / len(self.env_file_paths),
0, 1) 0, 1)
mean_reward = np.mean(self.simulation_rewards) mean_reward = round(np.mean(self.simulation_rewards), 2)
mean_normalized_reward = np.mean(self.simulation_rewards_normalized) mean_normalized_reward = round(np.mean(self.simulation_rewards_normalized), 2)
mean_percentage_complete = np.mean(self.simulation_percentage_complete) mean_percentage_complete = round(np.mean(self.simulation_percentage_complete), 3)
self.evaluation_state["state"] = "IN_PROGRESS" self.evaluation_state["state"] = "IN_PROGRESS"
self.evaluation_state["progress"] = progress self.evaluation_state["progress"] = progress
self.evaluation_state["simulation_count"] = self.simulation_count self.evaluation_state["simulation_count"] = self.simulation_count
...@@ -457,9 +457,9 @@ class FlatlandRemoteEvaluationService: ...@@ -457,9 +457,9 @@ class FlatlandRemoteEvaluationService:
""" """
) )
mean_reward = np.mean(self.simulation_rewards) mean_reward = round(np.mean(self.simulation_rewards), 2)
mean_normalized_reward = np.mean(self.simulation_rewards_normalized) mean_normalized_reward = round(np.mean(self.simulation_rewards_normalized), 2)
mean_percentage_complete = np.mean(self.simulation_percentage_complete) mean_percentage_complete = round(np.mean(self.simulation_percentage_complete), 3)
if self.visualize and len(os.listdir(self.vizualization_folder_name)) > 0: if self.visualize and len(os.listdir(self.vizualization_folder_name)) > 0:
# Generate the video # Generate the video
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment