Update param_pos

StatMixedML · Aug 28, 2023 · 7e7589c · 7e7589c
1 parent 8baa8d5
commit 7e7589c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightgbmlss/model.py b/lightgbmlss/model.py
@@ -502,7 +502,7 @@ def plot(self,
         explainer = shap.TreeExplainer(self.booster)
         shap_values = explainer(X)
 
-        param_pos = list(self.dist.param_dict.keys()).index(parameter)
+        param_pos = self.dist.distribution_arg_names.index(parameter)
 
         if plot_type == "Partial_Dependence":
             if self.dist.n_dist_param == 1: