v2-final: metrics

metrics.json

@@ -1,13 +1,13 @@
 {
-  "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,
+  "train_r2": 0.7732969260917075,
+  "test_r2": 0.7577273459547527,
+  "train_test_gap": 0.015569580136954864,
+  "r2": 0.7577273459547527,
+  "mae": 66.12724277172852,
+  "rmse": 86.85397055833653,
+  "cv_r2_mean": 0.761376771230841,
+  "cv_r2_std": 0.0036174258028472176,
+  "train_time_s": 0.37797045707702637,
   "features": [
     "num_packages",
     "total_weight_kg",
@@ -21,19 +21,18 @@
     "distance_per_stop"
   ],
   "feature_importances": {
-    "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
+    "num_packages": 0.15725062787532806,
+    "total_weight_kg": 0.01825214922428131,
+    "num_stops": 0.4727834463119507,
+    "avg_fragility": 0.01103278063237667,
+    "total_distance_km": 0.008022217079997063,
+    "route_difficulty_score": 0.25816723704338074,
+    "estimated_time_minutes": 0.04196315258741379,
+    "packages_per_stop": 0.02123946323990822,
+    "weight_per_package": 0.0068757119588553905,
+    "distance_per_stop": 0.004413126967847347
   },
   "train_samples": 20000,
   "test_samples": 5000,
-  "version": "v2",
-  "changes": "Realistic data with hidden confounders, heteroscedastic noise, non-linear interactions, measurement error, subpopulations"
+  "version": "v2-final"
 }