Upload model_metadata.json with huggingface_hub

model_metadata.json

@@ -28,18 +28,13 @@
   ],
   "n_features": 22,
   "target": "Engine Condition",
-  "threshold": 0.3162792985767373,
+  "threshold": 0.3699999999999999,
   "test_metrics": {
-    "balanced_accuracy": 0.5026111930544064,
-    "roc_auc": 0.6956703461204846,
-    "recall": 0.9978354978354979,
-    "precision": 0.631723192874272,
-    "f2": 0.8941906701580836
-  },
-  "test_metrics_at_default_threshold": {
-    "recall": 0.8463203463203464,
-    "precision": 0.6838653257542632,
-    "f2": 0.8079347039983469
+    "balanced_accuracy": 0.5159377210970009,
+    "roc_auc": 0.6980806781707058,
+    "recall": 0.9875541125541125,
+    "precision": 0.6381118881118881,
+    "f2": 0.8900702301989856
   },
   "best_params": {
     "base_max_depth": "3",
@@ -47,10 +42,8 @@
     "learning_rate": "0.2608158369431769"
   },
   "training_info": {
-    "cv_folds": 3,
-    "scoring": "balanced_accuracy",
-    "train_samples": 14651,
-    "val_samples": 1953,
+    "train_samples": 14650,
+    "val_samples": 1954,
     "test_samples": 2931
   }
 }