From 8315dcf03b665a7034668d4ccc009fd44507805e Mon Sep 17 00:00:00 2001
From: rogeriobonatti <rogeriobonatti@gmail.com>
Date: Fri, 12 Aug 2022 22:06:56 +0000
Subject: [PATCH] change

---
 mushr_rhc_ros/src/eval_all_models.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/mushr_rhc_ros/src/eval_all_models.py b/mushr_rhc_ros/src/eval_all_models.py
index 83dcb3a..e3a3431 100644
--- a/mushr_rhc_ros/src/eval_all_models.py
+++ b/mushr_rhc_ros/src/eval_all_models.py
@@ -62,7 +62,7 @@
     times = times[condition]
 
     # clear data above episode max
-    max_dist = 1500
+    max_dist = 1000
     condition = distances<max_dist
     distances = distances[condition]
     times = times[condition]
@@ -93,7 +93,7 @@
         '12L': all_vals_mean[2::4],
         '24L': all_vals_mean[3::4],
         # 'Dataset fraction': [0.0, 0.01, 0.1, 0.5, 1.0],
-        'Dataset fraction': ['540', '30K', '300K', '1.5M', '3M']}
+        'Number of Tokens Processed': ['540', '30K', '300K', '1.5M', '3M']}
 
 # data = {'3L':  all_vals_median[::4],
 #         '6L':  all_vals_median[1::4],
@@ -106,6 +106,8 @@
 
 df = pd.DataFrame(data)
 print(df)
-dfm = df.melt('Dataset fraction', var_name='cols', value_name='Average meters traveled')
-sns.catplot(x="Dataset fraction", y="Average meters traveled", hue='cols', data=dfm, kind='point')
+dfm = df.melt('Number of Tokens Processed', var_name='cols', value_name='Average meters traveled [m]')
+sns.catplot(x="Number of Tokens Processed", y="Average meters traveled [m]", hue='cols', data=dfm, kind='point')
+plt.grid()
+plt.gcf().subplots_adjust(bottom=0.15)
 plt.savefig('/home/azureuser/hackathon_data_premium/e2e_eval_models4/model_test/all_plots.png')
\ No newline at end of file