add filter in plotting

tongplw · Jul 3, 2020 · 1c48258 · 1c48258
1 parent 5594ac0
commit 1c48258
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tester.py b/tester.py
@@ -17,8 +17,8 @@
     for i in range(len(df)):
         waves = df.iloc[i].to_dict()
         attention = headset._attention(waves)
-        atts += [attention]
+        if attention < 1:
+            atts += [attention]
 
     plt.hist(atts, bins=200)
-    plt.show()
-
+    plt.show()