NPN / training_config.json
rishini's picture
fix: fill WRMSSE levels 1-5, regenerate parity fixtures (5k rows), recompute SHA256SUMS (part 2)
012ce04 verified
Raw
History Blame Contribute Delete
2.47 kB
{
"lgb_params": {
"objective": "tweedie",
"tweedie_variance_power": 1.1,
"metric": "rmse",
"num_leaves": 128,
"learning_rate": 0.03,
"min_data_in_leaf": 100,
"feature_fraction": 0.8,
"bagging_fraction": 0.8,
"bagging_freq": 1,
"n_estimators": 2000,
"early_stopping_rounds": 50,
"seed": 42,
"verbosity": -1,
"device": "gpu",
"num_threads": 4
},
"seed": 42,
"train_day_range": [
"d_1211",
"d_1941"
],
"versions": {
"python": "3.12.3",
"lightgbm": "4.5.0",
"numpy": "1.26.4",
"pandas": "2.2.3"
},
"platform": "Linux 6.11.0-1027-oem",
"git_commit": "none",
"local_wrmsse": {
"A": {
"total": 833.757607,
"state_id": 1183.48971,
"store_id": 416.204036,
"cat_id": 1314.256816,
"dept_id": 683.455299,
"state_cat": 523.703373,
"state_dept": 277.845457,
"store_cat": 191.934985,
"store_dept": 106.282299,
"item_id": 8.63059,
"item_state": 4.658677,
"item_store": 2.440745,
"overall": 148.327863
},
"B": {
"total": 678.952972,
"state_id": 788.99314,
"store_id": 277.469358,
"cat_id": 876.171211,
"dept_id": 455.636866,
"state_cat": 349.135582,
"state_dept": 185.230305,
"store_cat": 127.956656,
"store_dept": 70.854866,
"item_id": 5.753727,
"item_state": 3.105785,
"item_store": 1.627163,
"overall": 121.276794
},
"C": {
"total": 2441.678322,
"state_id": 939.277547,
"store_id": 330.320664,
"cat_id": 1043.060965,
"dept_id": 542.42484,
"state_cat": 415.637598,
"state_dept": 220.512268,
"store_cat": 152.329353,
"store_dept": 84.351031,
"item_id": 6.849675,
"item_state": 3.697363,
"item_store": 1.937099,
"overall": 167.073294
}
},
"wrmsse_levels": [
"total",
"state_id",
"store_id",
"cat_id",
"dept_id",
"state_cat",
"state_dept",
"store_cat",
"store_dept",
"item_id",
"item_state",
"item_store"
],
"evaluation_period": {
"fold_A": "2016-02-28 to 2016-03-27",
"fold_B": "2016-03-28 to 2016-04-24",
"fold_C": "2016-04-25 to 2016-05-22"
},
"aggregate_score": 145.559317,
"score_by_fold": {
"A": 148.327863,
"B": 121.276794,
"C": 167.073294
},
"parity_pred_sha256": "5f0b261b519c608137ffbf23ef312008ec14405804e750a575dfd6c464e7f70b"
}