From 8db0f2cf6c53997a3700bd80c78df650bffaef9e Mon Sep 17 00:00:00 2001
From: Ruben van de Ven <git@rubenvandeven.com>
Date: Tue, 31 Dec 2024 14:47:33 +0100
Subject: [PATCH] Tweak to have more trajectories to train from

---
 EXPERIMENTS/config.json | 8 ++++----
 trap/process_data.py    | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/EXPERIMENTS/config.json b/EXPERIMENTS/config.json
index 78e0876..4ef9b72 100644
--- a/EXPERIMENTS/config.json
+++ b/EXPERIMENTS/config.json
@@ -5,9 +5,9 @@
     "learning_rate": 0.01, 
     "min_learning_rate": 1e-05,
     "learning_decay_rate": 0.9999,
-    "prediction_horizon": 50,
-    "minimum_history_length": 10,
-    "maximum_history_length": 80,
+    "prediction_horizon": 30,
+    "minimum_history_length": 5,
+    "maximum_history_length": 50,
     "map_encoder": {
         "PEDESTRIAN": {
             "heading_state_index": [2, 3],
@@ -74,7 +74,7 @@
     "dynamic": {
         "PEDESTRIAN": {
             "name": "SingleIntegrator",
-            "distribution": false,
+            "distribution": true,
             "limits": {}
         }
     },
diff --git a/trap/process_data.py b/trap/process_data.py
index 474030f..2ff281c 100644
--- a/trap/process_data.py
+++ b/trap/process_data.py
@@ -145,7 +145,7 @@ def process_data(src_dir: Path, dst_dir: Path, name: str, smooth_tracks: bool, c
     print(max_frame_nr)
 
     # separate call so cursor is kept during multiple loops
-    seed(123)
+    # seed(123)
     shuffle(tracks)
 
     dt1 = RollingAverage()