feat: change percentage range

Lucs1590 · May 23, 2024 · 1a55cc6 · 1a55cc6
1 parent 4c6bfaa
commit 1a55cc6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main.py b/src/main.py
@@ -213,10 +213,10 @@ def preprocess_trackpoints_data(data):
     df = df.reset_index(drop=True)
     df = df.dropna()
 
-    if df.shape[0] > 1000:
+    if df.shape[0] > 4000:
+        df = run_euclidean_dist_deletion(df, 0.50)
+    elif df.shape[0] > 1000:
         df = run_euclidean_dist_deletion(df, 0.35)
-    elif df.shape[0] > 500:
-        df = run_euclidean_dist_deletion(df, 0.20)
     else:
         df = run_euclidean_dist_deletion(df, 0.10)