muthuk1's picture
v2-final: metrics
c9e74a0 verified
{
"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",
"num_stops",
"avg_fragility",
"total_distance_km",
"route_difficulty_score",
"estimated_time_minutes",
"packages_per_stop",
"weight_per_package",
"distance_per_stop"
],
"feature_importances": {
"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-final"
}