Dedulek commited on
Commit
41ebf30
·
verified ·
1 Parent(s): 0deed22

Upload rf_gaia_teff_corrected_log_20251126_130144_metadata.json with huggingface_hub

Browse files
rf_gaia_teff_corrected_log_20251126_130144_metadata.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "rf_gaia_teff_corrected_log_20251126_130144",
3
+ "timestamp": "20251126_130144",
4
+ "model_name": "Gaia \u2192 log(Corrected Teff)",
5
+ "description": "Temperature prediction using Gaia photometry on corrected and log-transformed Teff",
6
+ "model_type": "RandomForestRegressor",
7
+ "n_features": 6,
8
+ "features": [
9
+ "g",
10
+ "bp",
11
+ "rp",
12
+ "bp_rp",
13
+ "g_bp",
14
+ "g_rp"
15
+ ],
16
+ "data_source": "gaia_all_colors_teff_corrected.parquet",
17
+ "target": "teff_gaia_corrected_log",
18
+ "target_transform": "none",
19
+ "train_metrics": {
20
+ "mae": 0.03702315837236848,
21
+ "rmse": 0.05967808668708509,
22
+ "r2": 0.7285408623726284,
23
+ "mean_relative_error": 0.9789498319864837,
24
+ "within_5_percent": 98.16794850195704,
25
+ "within_10_percent": 99.92996595267853,
26
+ "within_20_percent": 100.0,
27
+ "n_samples": 1020932
28
+ },
29
+ "test_metrics": {
30
+ "mae": 0.041098999238153675,
31
+ "rmse": 0.06608836781262523,
32
+ "r2": 0.665321911952044,
33
+ "mean_relative_error": 1.0864054906608058,
34
+ "within_5_percent": 97.5054263930354,
35
+ "within_10_percent": 99.87775923270411,
36
+ "within_20_percent": 100.0,
37
+ "n_samples": 255234
38
+ },
39
+ "hyperparameters": {
40
+ "n_estimators": 300,
41
+ "max_depth": 20,
42
+ "min_samples_split": 10,
43
+ "min_samples_leaf": 5,
44
+ "max_features": "sqrt",
45
+ "random_state": 42,
46
+ "n_jobs": -1,
47
+ "verbose": 0
48
+ },
49
+ "config_file": "config/models/gaia_teff_corrected_log.yaml",
50
+ "data_config": {
51
+ "source_file": "gaia_all_colors_teff_corrected.parquet",
52
+ "source_format": "parquet",
53
+ "source_location": "processed",
54
+ "target": "teff_gaia_corrected_log",
55
+ "features": [
56
+ "g",
57
+ "bp",
58
+ "rp",
59
+ "bp_rp",
60
+ "g_bp",
61
+ "g_rp"
62
+ ],
63
+ "id_column": "source_id"
64
+ }
65
+ }