Upload model_metadata.json with huggingface_hub

model_metadata.json

@@ -1,5 +1,7 @@
 {
-  "model_name": "XGBoost",
+  "model_name": "Random Forest (Optuna)",
+  "algorithm": "Random Forest",
+  "tuning_method": "Optuna",
   "features": [
     "Engine RPM",
     "Lub Oil Pressure",
@@ -26,20 +28,21 @@
   ],
   "n_features": 22,
   "target": "Engine Condition",
-  "threshold": 0.14454752206802368,
+  "threshold": 0.5,
   "test_metrics": {
-    "balanced_accuracy": 0.6428076770874555,
-    "roc_auc": 0.6994469826879799,
-    "recall": 0.6309523809523809,
-    "precision": 0.7571428571428571,
-    "f2": 0.6527093596059114
+    "balanced_accuracy": 0.6455949982612033,
+    "roc_auc": 0.7011493046811607,
+    "recall": 0.6217532467532467,
+    "precision": 0.7624419376244194,
+    "f2": 0.6455781548488594
   },
   "best_params": {
-    "colsample_bytree": 0.8,
-    "learning_rate": 0.1,
-    "max_depth": 3,
-    "scale_pos_weight": 0.5859493396839143,
-    "subsample": 0.8
+    "n_estimators": "202",
+    "max_depth": "7",
+    "min_samples_split": "3",
+    "min_samples_leaf": "10",
+    "max_features": "sqrt",
+    "class_weight": "balanced_subsample"
   },
   "training_info": {
     "cv_folds": 3,