package-theft-accel / zcta_features_lightgbm_gpu_reg_metadata.json
LeonceNsh's picture
Upload 4 files
d78a5e7 verified
{
"model_type": "LGBMRegressor",
"training_metrics": {
"r2": 0.9933932362160545,
"mse": 0.00025389514099849084,
"rmse": 0.01593408739145392,
"y_pred": "[0.27489202 0.43708017 0.70934149 ... 0.46906955 0.41413552 0.40539368]"
},
"training_time": 1.042459487915039,
"feature_names": "['poverty_trend_slope' 'poverty_volatility'\n 'poverty_coefficient_variation' 'poverty_acceleration'\n 'poverty_recent_momentum' 'poverty_skewness' 'poverty_kurtosis'\n 'poverty_change_point_strength' 'poverty_min' 'poverty_max' 'latitude'\n 'longitude' 'urbanity_score' 'embedding_cluster_distance'\n 'poverty_risk_2022_x_urbanity_score'\n 'poverty_risk_2022_ratio_urbanity_score' 'population' 'log_population'\n 'sqrt_population' 'cbrt_population' 'population_density_proxy'\n 'poverty_risk_2022_x_log_population' 'pca_embedding_0' 'pca_embedding_1'\n 'pca_embedding_2' 'pca_embedding_3' 'pca_embedding_4' 'pca_embedding_5'\n 'pca_embedding_6' 'pca_embedding_7' 'pca_embedding_8' 'pca_embedding_9'\n '2011' '2012' '2013' '2014' '2015' '2016' '2017' '2018' '2019' '2020'\n '2021' '2022' 'feature0' 'feature1' 'feature2' 'feature3' 'feature4'\n 'feature5']",
"model_params": {
"boosting_type": "gbdt",
"class_weight": null,
"colsample_bytree": 1.0,
"importance_type": "split",
"learning_rate": 0.1,
"max_depth": 6,
"min_child_samples": 20,
"min_child_weight": 0.001,
"min_split_gain": 0.0,
"n_estimators": 200,
"n_jobs": null,
"num_leaves": 31,
"objective": null,
"random_state": 42,
"reg_alpha": 0.0,
"reg_lambda": 0.0,
"subsample": 1.0,
"subsample_for_bin": 200000,
"subsample_freq": 0,
"device": "gpu",
"gpu_platform_id": 0,
"gpu_device_id": 0,
"verbosity": -1
},
"created_at": "2025-07-24T22:51:00.955700",
"gpu_trained": true
}