Update train_model_mixin.py

sgoldenlab · Oct 10, 2023 · 3b5003b · 3b5003b
1 parent 904eb5f
commit 3b5003b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/simba/mixins/train_model_mixin.py b/simba/mixins/train_model_mixin.py
@@ -771,7 +771,9 @@ def create_shap_log(
            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))$$
+           \begin{equation}
+           \text{Mobility} = \sqrt{\frac{\text{variance of the first derivative}}{\text{variance of the signal}}}
+           \end{equation}
 
         :param str ini_file_path: Path to the SimBA project_config.ini
         :param RandomForestClassifier rf_clf: sklearn random forest classifier