From fe5c654a278b9c31aede8a56265c004dd370aa08 Mon Sep 17 00:00:00 2001
From: Andi Gerken <andi.gerken@gmail.com>
Date: Tue, 29 Dec 2020 12:03:35 +0100
Subject: [PATCH] Minor evaluation development

---
 src/robofish/evaluate/evaluate.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/robofish/evaluate/evaluate.py b/src/robofish/evaluate/evaluate.py
index 6dfb7e9..730a47e 100644
--- a/src/robofish/evaluate/evaluate.py
+++ b/src/robofish/evaluate/evaluate.py
@@ -20,11 +20,19 @@ def evaluate_speed(paths):
 
         speeds.append(path_speeds)
 
+    print(len(speeds))
+    max_len = max([len(speed) for speed in speeds])
+
+    print(max_len)
+    print([len(speed) for speed in speeds])
+    # speeds = [speed[:max_len] for speed in speeds]
+
     plt.hist(speeds, bins=20, label=paths, density=True, range=[0, 1])
-    # sns.displot(speeds, label=paths, multiple="layer", kind="kde")
+    # sns.displot(speeds, label=paths, stat="probability", kind="hist")
+    # sns.displot(speeds[1], label=paths[1], stat="probability", kind="hist")
     plt.title("Agent speeds")
     plt.xlabel("Speed (cm/timestep)")
-    plt.ylabel("Frequency (logscale)")
+    plt.ylabel("Frequency")
     plt.ticklabel_format(useOffset=False)
     # plt.xscale("log", nonpositive="clip")
     plt.legend()
-- 
GitLab