Commit 67b0a673 authored by Aravind Balakrishnan's avatar Aravind Balakrishnan

fixed error with mean and std

parent 9c5bcb29
......@@ -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),
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment