Skip to content

Commit 67978af

Browse files
committed
remove importance in XGBoost Fit since we have Explainer Dashboard
1 parent 548dac4 commit 67978af

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

README.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@ Here's an example of how to use the library to run an XGBoost pipeline:
4646
```json
4747
{
4848
"custom_steps_path": "examples/ocf/",
49-
"save_path": "runs/xgboost_train.pkl",
5049
"pipeline": {
5150
"name": "XGBoostTrainingPipeline",
5251
"description": "Training pipeline for XGBoost models.",

pipeline_lib/implementation/tabular/xgboost/fit_model.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,10 +90,6 @@ def execute(self, data: DataContainer) -> DataContainer:
9090
# Save the model to the data container
9191
data[DataContainer.MODEL] = model
9292

93-
importance = model.get_booster().get_score(importance_type="gain")
94-
importance = dict(sorted(importance.items(), key=lambda item: item[1], reverse=True))
95-
data[DataContainer.IMPORTANCE] = importance
96-
9793
# save model to disk
9894
save_path = self.save_path
9995

0 commit comments

Comments
 (0)