tweak for when base_score is None

oegedijk · Feb 11, 2023 · 69e5552 · 69e5552
1 parent bfefde5
commit 69e5552
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/explainerdashboard/explainer_methods.py b/explainerdashboard/explainer_methods.py
@@ -1439,9 +1439,9 @@ def get_xgboost_preds_df(xgbmodel, X_row, pos_label=1):
         n_classes = len(xgbmodel.classes_)
         if n_classes == 2:
             if pos_label==1:
-                base_proba = xgbmodel.get_params()['base_score']
+                base_proba = xgbmodel.get_params()['base_score'] or 0.5
             elif pos_label==0:
-                base_proba = 1 - xgbmodel.get_params()['base_score']
+                base_proba = 1 - xgbmodel.get_params()['base_score'] or 0.5
             else:
                 raise ValueError("pos_label should be either 0 or 1!")
             n_trees = len(xgbmodel.get_booster().get_dump())