Upload rf_gaia_colors_flag1_20251218_094937_metadata.json with huggingface_hub

rf_gaia_colors_flag1_20251218_094937_metadata.json

@@ -0,0 +1,74 @@
+{
+  "model_id": "rf_gaia_colors_flag1_20251218_094937",
+  "timestamp": "20251218_094937",
+  "model_name": "Gaia Colors Only \u2192 Corrected Teff (Flag 1 High Quality)",
+  "description": "Temperature prediction from Gaia photometry only, trained on high-quality flag 1 sources with corrected Teff",
+  "model_type": "RandomForestRegressor",
+  "n_features": 6,
+  "features": [
+    "g",
+    "bp",
+    "rp",
+    "bp_rp",
+    "g_rp",
+    "g_bp"
+  ],
+  "data_source": "eb_unified_photometry_with_flags.parquet",
+  "target": "teff_gaia_corrected",
+  "target_transform": "none",
+  "train_metrics": {
+    "mae": 163.53359735932554,
+    "rmse": 251.54784298582464,
+    "r2": 0.9423517422227697,
+    "mean_relative_error": 3.4108434410762993,
+    "within_5_percent": 76.03296494839566,
+    "within_10_percent": 93.90267892899047,
+    "within_20_percent": 99.84613547104327,
+    "n_samples": 569332
+  },
+  "test_metrics": {
+    "mae": 195.47009591486,
+    "rmse": 311.6045853674034,
+    "r2": 0.9126003555477147,
+    "mean_relative_error": 4.047084479869949,
+    "within_5_percent": 70.83479702671181,
+    "within_10_percent": 90.65648404457122,
+    "within_20_percent": 99.49414756839546,
+    "n_samples": 142334
+  },
+  "hyperparameters": {
+    "n_estimators": 300,
+    "max_depth": 20,
+    "min_samples_split": 5,
+    "min_samples_leaf": 4,
+    "max_features": "log2",
+    "random_state": 42,
+    "n_jobs": -1,
+    "verbose": 0
+  },
+  "config_file": "config/models/gaia_colors_only_flag1.yaml",
+  "data_config": {
+    "source_file": "eb_unified_photometry_with_flags.parquet",
+    "source_format": "parquet",
+    "source_location": "processed",
+    "target": "teff_gaia_corrected",
+    "features": [
+      "g",
+      "bp",
+      "rp",
+      "bp_rp",
+      "g_rp",
+      "g_bp"
+    ],
+    "id_column": "source_id"
+  },
+  "teff_correction": {
+    "n_corrected": 19681,
+    "mean_correction": 2947.2440219483956,
+    "median_correction": 2641.1160991821525,
+    "threshold": 10000,
+    "degree": 2,
+    "target_column": "teff_gaia",
+    "corrected_column": "teff_gaia_corrected"
+  }
+}