From 67b0a67356713797267b99bdbc89289a2fb43cf7 Mon Sep 17 00:00:00 2001
From: Aravind Balakrishnan <aravind.balakrishnan@uwaterloo.ca>
Date: Tue, 26 Mar 2019 17:49:21 -0400
Subject: [PATCH] fixed error with mean and std
---
high_level_policy_main.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/high_level_policy_main.py b/high_level_policy_main.py
index e24b8e9..45893b4 100644
--- a/high_level_policy_main.py
+++ b/high_level_policy_main.py
@@ -129,6 +129,9 @@ def evaluate_high_level_policy(nb_episodes_for_test=100,
print("Termination reason(s):")
for reason, count_list in termination_reason_list.items():
count_list = np.array(count_list)
+ while count_list.size != nb_trials:
+ count_list = np.append(count_list,0)
+
print("{}: Avg: {}, Std: {}".format(reason, np.mean(count_list),
np.std(count_list)))
--
GitLab