Update README.md
Browse files
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/
|
| 82 |
-
xgb_model = joblib.load(hf_hub_download("AurelPx/
|
| 83 |
-
encoder = joblib.load(hf_hub_download("AurelPx/
|
| 84 |
cb_model = cb.CatBoostClassifier()
|
| 85 |
-
cb_model.load_model(hf_hub_download("AurelPx/
|
| 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/
|
| 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 |
```
|