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