vimp dropna fix

vnijs · Nov 27, 2024 · dbb4356 · dbb4356
1 parent 3544448
commit dbb4356
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyrsm/model/visualize.py b/pyrsm/model/visualize.py
@@ -535,7 +535,7 @@ def vimp_plot_sm(fitted, data, rep=10, ax=None, ret=False):
 
     rvar = extract_rvar(model, data.columns)
     evars = extract_evars(model, data.columns)
-    data = data[[rvar] + evars].copy().reset_index(drop=True).dropna()
+    data = data[[rvar] + evars].copy().dropna().reset_index(drop=True)
 
     if len(model.endog) != data.shape[0]:
         raise Exception(
@@ -578,6 +578,7 @@ def imp_calc_logit(base, pred):
     importance_values = {k: [v / rep] for k, v in importance_values.items()}
     sorted_idx = pd.DataFrame(importance_values).transpose()
     sorted_idx = sorted_idx.sort_values(0, ascending=True)
+    print(importance_values)
     fig = sorted_idx.plot.barh(
         color="slateblue",
         legend=None,