Skip to content

Commit

Permalink
fixed pdp drop_na bug
Browse files Browse the repository at this point in the history
  • Loading branch information
oegedijk committed Dec 8, 2020
1 parent 54bea98 commit 639e45d
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 7 deletions.
2 changes: 1 addition & 1 deletion RELEASE_NOTES.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
stored parameters, e.g. `db = ExplainerDashboard.from_config("dashboard.yaml", higher_is_better=False)`

### Bug Fixes
-
- fixed bug where `drop_na=True` for `explainer.plot_pdp()` was not working.
-

### Improvements
Expand Down
5 changes: 0 additions & 5 deletions TODO.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,6 @@


### Classifier plots:
- add label percentage at cutoff to cumulative precision plot

- add wizard to lift curve
- pdp: add multiclass option
- no icelines to keep it from getting too busy?
Expand Down Expand Up @@ -62,7 +60,6 @@
- generate groups programmatically!

### Components
- add description to PredictionSummaryComponent (reg+clas)
- add hide_footer to components with a CardFooter

- add querystring method to ExplainerComponents
Expand All @@ -88,8 +85,6 @@
- Add this method? : https://arxiv.org/abs/2006.04750?

## Tests:
- add prediction_summary_df test

- test model_output='probability' and 'raw' or 'logodds' seperately
- write tests for explainer_methods
- write tests for explainer_plots
Expand Down
2 changes: 1 addition & 1 deletion explainerdashboard/explainers.py
Original file line number Diff line number Diff line change
Expand Up @@ -1525,7 +1525,7 @@ def plot_pdp(self, col, index=None, X_row=None, drop_na=True, sample=100,
"""
pdp_result = self.get_pdp_result(col, index, X_row,
num_grid_points=gridpoints, pos_label=pos_label)
drop_na=drop_na, sample=sample, num_grid_points=gridpoints, pos_label=pos_label)
units = "Predicted %" if self.model_output=='probability' else self.units
if index is not None:
col_value, pred = self.get_col_value_plus_prediction(col, index=index, pos_label=pos_label)
Expand Down

0 comments on commit 639e45d

Please sign in to comment.