Dedulek commited on
Commit
87646eb
·
verified ·
1 Parent(s): b8bed8c

Upload rf_gaia_2mass_ir_20251103_141119_metadata.json with huggingface_hub

Browse files
rf_gaia_2mass_ir_20251103_141119_metadata.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "rf_gaia_2mass_ir_20251103_141119",
3
+ "timestamp": "20251103_141119",
4
+ "model_name": "Gaia + 2MASS IR",
5
+ "description": "Temperature prediction using Gaia optical and 2MASS infrared photometry",
6
+ "model_type": "RandomForestRegressor",
7
+ "n_features": 5,
8
+ "features": [
9
+ "phot_g_mean_mag",
10
+ "bp_rp",
11
+ "j_h_color",
12
+ "h_k_color",
13
+ "j_k_color"
14
+ ],
15
+ "data_source": "gaia_2mass_ir_train.parquet",
16
+ "target": "teff_gspphot",
17
+ "train_metrics": {
18
+ "mae": 466.3848224965843,
19
+ "rmse": 761.6802111914476,
20
+ "r2": 0.7468184080404487,
21
+ "mean_relative_error": 7.75831024412814,
22
+ "within_5_percent": 45.870779750213096,
23
+ "within_10_percent": 73.05849979616795,
24
+ "within_20_percent": 92.91313048956751,
25
+ "n_samples": 431728
26
+ },
27
+ "test_metrics": {
28
+ "mae": 588.8477953125182,
29
+ "rmse": 965.4933618203581,
30
+ "r2": 0.5930834905512652,
31
+ "mean_relative_error": 9.758310497789818,
32
+ "within_5_percent": 38.448282251788164,
33
+ "within_10_percent": 64.79172071304154,
34
+ "within_20_percent": 88.06100137123374,
35
+ "n_samples": 107932
36
+ },
37
+ "hyperparameters": {
38
+ "n_estimators": 300,
39
+ "max_depth": 20,
40
+ "min_samples_split": 5,
41
+ "min_samples_leaf": 4,
42
+ "max_features": "log2",
43
+ "random_state": 42,
44
+ "n_jobs": -1,
45
+ "verbose": 0
46
+ },
47
+ "config_file": "config/models/gaia_2mass_ir.yaml"
48
+ }