Skip to content

Commit

Permalink
Update train_model_mixin.py
Browse files Browse the repository at this point in the history
  • Loading branch information
sronilsson authored Oct 10, 2023
1 parent 3fa061b commit 64bb99a
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions simba/mixins/train_model_mixin.py
Original file line number Diff line number Diff line change
Expand Up @@ -766,15 +766,15 @@ def create_shap_log(
.. image:: _static/img/shap.png
:width: 400
:align: center
.. math::
\phi_i(f, x) = \\sum_{S \\subseteq F \\setminus {i}} \\frac{|S|!(|F| - |S| - 1)!}{|F|!} (f_{S \cup {i}}(x_{S \\cup {i}}) - f_S(x_S))
.. note::
For improved run-times, use multiprocessing through :meth:`simba.mixins.train_model_mixins.TrainModelMixin.create_shap_log_mp`
.. math::
:align: center
\phi_i(f, x) = \\sum_{S \\subseteq F \\setminus {i}} \\frac{|S|!(|F| - |S| - 1)!}{|F|!} (f_{S \cup {i}}(x_{S \\cup {i}}) - f_S(x_S))
:param str ini_file_path: Path to the SimBA project_config.ini
:param RandomForestClassifier rf_clf: sklearn random forest classifier
Expand Down

0 comments on commit 64bb99a

Please sign in to comment.