Update train_model_mixin.py

sgoldenlab · Oct 10, 2023 · 3010e9a · 3010e9a
1 parent 53b0fe5
commit 3010e9a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/simba/mixins/train_model_mixin.py b/simba/mixins/train_model_mixin.py
@@ -770,6 +770,9 @@ def create_shap_log(
         .. note::
            For improved run-times, use multiprocessing through :meth:`simba.mixins.train_model_mixins.TrainModelMixin.create_shap_log_mp`
 
+        .. 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))$$
+
         :param str ini_file_path: Path to the SimBA project_config.ini
         :param RandomForestClassifier rf_clf: sklearn random forest classifier
         :param pd.DataFrame x_df: Test features.