Skip to content

Commit

Permalink
Update chart names (#118)
Browse files Browse the repository at this point in the history
  • Loading branch information
HubertJaworski authored Nov 23, 2020
1 parent 3d4477a commit 839ad07
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions neptunecontrib/api/explainers.py
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 +209,7 @@ def log_global_explanations(explainer, categorical_features=None, numerical_feat
ale_num.result["_label_"] = 'ale'

pdp_num_plot = pdp_num.plot(ale_num, show=False)
log_chart(name="Partial Dependence", chart=pdp_num_plot, experiment=_exp)
log_chart(name="Aggregated Profiles Numerical", chart=pdp_num_plot, experiment=_exp)

if categorical_features:
pdp_cat = explainer.model_profile(type='partial', variable_type='categorical', variables=categorical_features)
Expand All @@ -220,4 +220,4 @@ def log_global_explanations(explainer, categorical_features=None, numerical_feat
ale_cat.result['_label_'] = 'ale'

ale_cat_plot = ale_cat.plot(pdp_cat, show=False)
log_chart(name="Accumulated Dependence", chart=ale_cat_plot, experiment=_exp)
log_chart(name="Aggregated Profiles Categorical", chart=ale_cat_plot, experiment=_exp)

0 comments on commit 839ad07

Please sign in to comment.