Tabular Regression
Scikit-learn
logistics
xgboost
delivery
driver-effort
fairrelay
muthuk1 commited on
Commit
2286f5c
·
verified ·
1 Parent(s): a33a7c4

Add metrics for Driver Effort Prediction Model

Browse files
Files changed (1) hide show
  1. metrics.json +30 -0
metrics.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "mae": 3.6631188250223796,
3
+ "rmse": 6.95660019937431,
4
+ "r2": 0.9702407057933086,
5
+ "cv_r2_mean": 0.9732488798210385,
6
+ "cv_r2_std": 0.0016249976898991066,
7
+ "train_time_s": 0.4539225101470947,
8
+ "features": [
9
+ "num_packages",
10
+ "total_weight_kg",
11
+ "num_stops",
12
+ "route_difficulty_score",
13
+ "estimated_time_minutes",
14
+ "experience_days",
15
+ "recent_avg_workload",
16
+ "recent_hard_days"
17
+ ],
18
+ "feature_importances": {
19
+ "num_packages": 0.2921975553035736,
20
+ "total_weight_kg": 0.08021678775548935,
21
+ "num_stops": 0.07934663444757462,
22
+ "route_difficulty_score": 0.005875657312572002,
23
+ "estimated_time_minutes": 0.4705999195575714,
24
+ "experience_days": 0.060129713267087936,
25
+ "recent_avg_workload": 0.002936065196990967,
26
+ "recent_hard_days": 0.008697610348463058
27
+ },
28
+ "train_samples": 12000,
29
+ "test_samples": 3000
30
+ }