v2: updated metrics

metrics.json

@@ -1,10 +1,13 @@
 {
-  "mae": 4.233838771311442,
-  "rmse": 5.341754710278607,
-  "r2": 0.9969348810311977,
-  "cv_r2_mean": 0.9968869649451856,
-  "cv_r2_std": 0.00010854830744197988,
-  "train_time_s": 0.2467200756072998,
+  "train_r2": 0.8281628859803508,
+  "test_r2": 0.7535808096405573,
+  "train_test_gap": 0.07458207633979352,
+  "mae": 66.52149847943116,
+  "rmse": 87.59407722603619,
+  "r2": 0.7535808096405573,
+  "cv_r2_mean": 0.7582231114724559,
+  "cv_r2_std": 0.003137890788865629,
+  "train_time_s": 0.7617073059082031,
   "features": [
     "num_packages",
     "total_weight_kg",
@@ -12,17 +15,25 @@
     "avg_fragility",
     "total_distance_km",
     "route_difficulty_score",
-    "estimated_time_minutes"
+    "estimated_time_minutes",
+    "packages_per_stop",
+    "weight_per_package",
+    "distance_per_stop"
   ],
   "feature_importances": {
-    "num_packages": 0.02053593471646309,
-    "total_weight_kg": 0.01176395732909441,
-    "num_stops": 0.3539225161075592,
-    "avg_fragility": 0.003524457337334752,
-    "total_distance_km": 0.0008101653074845672,
-    "route_difficulty_score": 0.545396625995636,
-    "estimated_time_minutes": 0.06404635310173035
+    "num_packages": 0.0373411551117897,
+    "total_weight_kg": 0.016120798885822296,
+    "num_stops": 0.7063920497894287,
+    "avg_fragility": 0.014484221115708351,
+    "total_distance_km": 0.01030807476490736,
+    "route_difficulty_score": 0.17422015964984894,
+    "estimated_time_minutes": 0.010052479803562164,
+    "packages_per_stop": 0.010061535984277725,
+    "weight_per_package": 0.01162794977426529,
+    "distance_per_stop": 0.009391536004841328
   },
-  "train_samples": 9600,
-  "test_samples": 2400
+  "train_samples": 20000,
+  "test_samples": 5000,
+  "version": "v2",
+  "changes": "Realistic data with hidden confounders, heteroscedastic noise, non-linear interactions, measurement error, subpopulations"
 }