nba-performance-predictor / nba_model /model_metadata.json
arvindrangarajan's picture
Upload nba_model/model_metadata.json with huggingface_hub
b8be8f8 verified
{
"feature_names": [
"Age",
"G",
"GS",
"MP",
"FG",
"FGA",
"FG_1",
"3P",
"3PA",
"3P_1",
"2P",
"2PA",
"2P_1",
"eFG",
"FT",
"FTA",
"FT_1",
"ORB",
"DRB",
"TRB",
"AST",
"STL",
"BLK",
"TOV",
"PF",
"Pos_encoded",
"Team_encoded",
"Age_category_encoded",
"Points_per_minute",
"Efficiency",
"Usage_rate",
"PTS_lag_1",
"PTS_lag_2",
"PTS_lag_3",
"TRB_lag_1",
"TRB_lag_2",
"TRB_lag_3",
"AST_lag_1",
"AST_lag_2",
"AST_lag_3",
"FG_1_lag_1",
"FG_1_lag_2",
"FG_1_lag_3",
"FT_1_lag_1",
"FT_1_lag_2",
"FT_1_lag_3",
"MP_lag_1",
"MP_lag_2",
"MP_lag_3",
"STL_lag_1",
"STL_lag_2",
"STL_lag_3",
"BLK_lag_1",
"BLK_lag_2",
"BLK_lag_3",
"Points_per_minute_lag_1",
"Points_per_minute_lag_2",
"Points_per_minute_lag_3",
"Efficiency_lag_1",
"Efficiency_lag_2",
"Efficiency_lag_3",
"Usage_rate_lag_1",
"Usage_rate_lag_2",
"Usage_rate_lag_3",
"PTS_rolling_2",
"PTS_rolling_3",
"TRB_rolling_2",
"TRB_rolling_3",
"AST_rolling_2",
"AST_rolling_3",
"FG_1_rolling_2",
"FG_1_rolling_3",
"FT_1_rolling_2",
"FT_1_rolling_3",
"MP_rolling_2",
"MP_rolling_3",
"STL_rolling_2",
"STL_rolling_3",
"BLK_rolling_2",
"BLK_rolling_3",
"Points_per_minute_rolling_2",
"Points_per_minute_rolling_3",
"Efficiency_rolling_2",
"Efficiency_rolling_3",
"Usage_rate_rolling_2",
"Usage_rate_rolling_3",
"PTS_trend",
"TRB_trend",
"AST_trend",
"FG_1_trend",
"FT_1_trend",
"MP_trend",
"STL_trend",
"BLK_trend",
"Points_per_minute_trend",
"Efficiency_trend",
"Usage_rate_trend"
],
"target_column": "PTS",
"model_type": "XGBRegressor",
"model_params": {
"objective": "reg:squarederror",
"base_score": null,
"booster": null,
"callbacks": null,
"colsample_bylevel": null,
"colsample_bynode": null,
"colsample_bytree": 1.0,
"device": null,
"early_stopping_rounds": null,
"enable_categorical": false,
"eval_metric": null,
"feature_types": null,
"feature_weights": null,
"gamma": null,
"grow_policy": null,
"importance_type": null,
"interaction_constraints": null,
"learning_rate": 0.1,
"max_bin": null,
"max_cat_threshold": null,
"max_cat_to_onehot": null,
"max_delta_step": null,
"max_depth": 3,
"max_leaves": null,
"min_child_weight": null,
"missing": NaN,
"monotone_constraints": null,
"multi_strategy": null,
"n_estimators": 300,
"n_jobs": -1,
"num_parallel_tree": null,
"random_state": 42,
"reg_alpha": null,
"reg_lambda": null,
"sampling_method": null,
"scale_pos_weight": null,
"subsample": 1.0,
"tree_method": null,
"validate_parameters": null,
"verbosity": null
}
}