remove importance in XGBoost Fit since we have Explainer Dashboard

tryolabs · Mar 14, 2024 · 67978af · 67978af
1 parent 548dac4
commit 67978af
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -46,7 +46,6 @@ Here's an example of how to use the library to run an XGBoost pipeline:
 ```json
 {
     "custom_steps_path": "examples/ocf/",
-    "save_path": "runs/xgboost_train.pkl",
     "pipeline": {
         "name": "XGBoostTrainingPipeline",
         "description": "Training pipeline for XGBoost models.",

diff --git a/pipeline_lib/implementation/tabular/xgboost/fit_model.py b/pipeline_lib/implementation/tabular/xgboost/fit_model.py
@@ -90,10 +90,6 @@ def execute(self, data: DataContainer) -> DataContainer:
         # Save the model to the data container
         data[DataContainer.MODEL] = model
 
-        importance = model.get_booster().get_score(importance_type="gain")
-        importance = dict(sorted(importance.items(), key=lambda item: item[1], reverse=True))
-        data[DataContainer.IMPORTANCE] = importance
-
         # save model to disk
         save_path = self.save_path