use only predicted prob of class 1

Author: bdwilliamson

vimpy/predictiveness_measures.py

@@ -65,7 +65,7 @@ def cv_predictiveness(x, y, S, measure, pred_func, V = 5, stratified = True, na_
                 if measure.__name__ in ["r_squared"]:
                     preds_v = pred_func.predict(x_test[:, S])
                 else:
-                    preds_v = pred_func.predict_proba(x_test[:, S])
+                    preds_v = pred_func.predict_proba(x_test[:, S])[:, 1]
             preds[cc_cond[fold_cond]] = preds_v
             vs[v] = measure(y_test, preds_v)
             ics[cc_cond[fold_cond]] = compute_ic(y_test, preds_v, measure.__name__)

vimpy/spvim.py

@@ -4,9 +4,10 @@
 ## import required libraries
 import numpy as np
 from scipy.stats import norm
-import predictiveness_measures as mp
-import spvim_ic as sic
-import utils as uts
+from .predictiveness_measures import cv_predictiveness
+from .spvim_ic import shapley_influence_function, shapley_se
+from .vimpy_utils import get_measure_function
+
 
 class spvim:
 

vimpy/vim.py

@@ -5,7 +5,7 @@
 import numpy as np
 from scipy.stats import norm
 from .predictiveness_measures import cv_predictiveness, cv_predictiveness_precomputed
-from .vimpy_utils import get_measure_function, make_folds
+from .vimpy_utils import get_measure_function
 
 
 class vim: