AurelPx commited on
Commit
0ba0075
·
verified ·
1 Parent(s): 66cd2ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -78,11 +78,11 @@ Mean 0.9313 ± 0.0029
78
  import joblib, catboost as cb
79
  from huggingface_hub import hf_hub_download
80
 
81
- lgb_model = joblib.load(hf_hub_download("AurelPx/IncomeSlayer-9000", "lgb_model.pkl"))
82
- xgb_model = joblib.load(hf_hub_download("AurelPx/IncomeSlayer-9000", "xgb_model.pkl"))
83
- encoder = joblib.load(hf_hub_download("AurelPx/IncomeSlayer-9000", "ordinal_encoder.pkl"))
84
  cb_model = cb.CatBoostClassifier()
85
- cb_model.load_model(hf_hub_download("AurelPx/IncomeSlayer-9000", "cb_model.cbm"))
86
 
87
  # Build X_enc (28 features) and X_cb_df (21 cols, native categoricals) — see train.py
88
  proba = 0.1 * lgb_model.predict_proba(X_enc)[:, 1] \
@@ -115,6 +115,6 @@ Full preprocessing pipeline in `train.py`.
115
  author = {AurelPx},
116
  title = {Stacked GBM Ensemble for Income Classification (OpenML Task 7592)},
117
  year = {2026},
118
- url = {https://huggingface.co/AurelPx/IncomeSlayer-9000}
119
  }
120
  ```
 
78
  import joblib, catboost as cb
79
  from huggingface_hub import hf_hub_download
80
 
81
+ lgb_model = joblib.load(hf_hub_download("AurelPx/BoostingEnsemble-Income-Classification", "lgb_model.pkl"))
82
+ xgb_model = joblib.load(hf_hub_download("AurelPx/BoostingEnsemble-Income-Classification", "xgb_model.pkl"))
83
+ encoder = joblib.load(hf_hub_download("AurelPx/BoostingEnsemble-Income-Classification", "ordinal_encoder.pkl"))
84
  cb_model = cb.CatBoostClassifier()
85
+ cb_model.load_model(hf_hub_download("AurelPx/BoostingEnsemble-Income-Classification", "cb_model.cbm"))
86
 
87
  # Build X_enc (28 features) and X_cb_df (21 cols, native categoricals) — see train.py
88
  proba = 0.1 * lgb_model.predict_proba(X_enc)[:, 1] \
 
115
  author = {AurelPx},
116
  title = {Stacked GBM Ensemble for Income Classification (OpenML Task 7592)},
117
  year = {2026},
118
+ url = {https://huggingface.co/AurelPx/BoostingEnsemble-Income-Classification}
119
  }
120
  ```