Dedulek commited on
Commit
5f90b84
·
verified ·
1 Parent(s): 5802627

Upload rf_gaia_all_colors_teff_log_20251112_162857_metadata.json with huggingface_hub

Browse files
rf_gaia_all_colors_teff_log_20251112_162857_metadata.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "rf_gaia_all_colors_teff_log_20251112_162857",
3
+ "timestamp": "20251112_162857",
4
+ "model_name": "Gaia All Colors \u2192 log(Teff)",
5
+ "description": "Temperature prediction using log-transformed target and all Gaia colors (BP-RP, G-BP, G-RP) and bands",
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_train.parquet",
17
+ "target": "teff_gaia",
18
+ "target_transform": "log",
19
+ "train_metrics": {
20
+ "mae": 484.4404060985601,
21
+ "rmse": 895.6126476114173,
22
+ "r2": 0.7260720781309906,
23
+ "mean_relative_error": 7.874726994435164,
24
+ "within_5_percent": 50.58603315402005,
25
+ "within_10_percent": 72.69602676769853,
26
+ "within_20_percent": 91.01928434019112,
27
+ "n_samples": 1020932
28
+ },
29
+ "test_metrics": {
30
+ "mae": 556.9184842625981,
31
+ "rmse": 1021.2615423937236,
32
+ "r2": 0.6404160611505488,
33
+ "mean_relative_error": 9.02638564896777,
34
+ "within_5_percent": 46.35785201031211,
35
+ "within_10_percent": 68.47246056559862,
36
+ "within_20_percent": 88.57166364982722,
37
+ "n_samples": 255234
38
+ },
39
+ "hyperparameters": {
40
+ "n_estimators": 300,
41
+ "max_depth": 20,
42
+ "min_samples_split": 5,
43
+ "min_samples_leaf": 2,
44
+ "max_features": "sqrt",
45
+ "random_state": 42,
46
+ "n_jobs": -1,
47
+ "verbose": 0
48
+ },
49
+ "config_file": "config/models/gaia_all_colors_teff_log.yaml",
50
+ "data_config": {
51
+ "source_file": "gaia_all_colors_train.parquet",
52
+ "source_format": "parquet",
53
+ "source_location": "processed",
54
+ "target": "teff_gaia",
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
+ }