Skip to content

Commit dfbf9bb

Browse files
committed
adds set_shap_values to explainer docs
1 parent d6a3666 commit dfbf9bb

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

docs/source/explainers.rst

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,16 @@ You can set the precision of the calculated shap values, predictions, etc, in
258258
order to save on memory usage. Default is ``'float64'``, but ``'float32'`` is probably
259259
fine, maybe even ``'float16'`` for your application.
260260

261+
Pre-calculated shap values
262+
==========================
263+
264+
Perhaps you already have calculated the shap values somewhere, or you can calculate
265+
them off on a giant cluster somewhere, or your model supports `GPU generated shap values <https://github.com/rapidsai/gputreeshap>`_.
266+
267+
You can simply add these pre-calculated shap values to the explainer with
268+
``explainer.set_shap_values()`` and ``explainer.set_shap_interaction_values()`` methods.
269+
270+
261271
Plots
262272
=====
263273

0 commit comments

Comments
 (0)