Merge pull request #39 from stat17-hb/master

Update skope_rules.py (fix invalid mask indexing for OOB samples)

Author: ngoix

skrules/skope_rules.py

@@ -7,6 +7,7 @@
 from sklearn.base import BaseEstimator
 from sklearn.utils.validation import check_X_y, check_array, check_is_fitted
 from sklearn.utils.multiclass import check_classification_targets
+from sklearn.utils import indices_to_mask
 from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
 from sklearn.ensemble import BaggingClassifier, BaggingRegressor
 from sklearn.tree import _tree
@@ -333,7 +334,8 @@ def fit(self, X, y, sample_weight=None):
                                                 self.estimators_features_):
 
             # Create mask for OOB samples
-            mask = ~samples
+            mask = ~indices_to_mask(samples, n_samples)            
+                        
             if sum(mask) == 0:
                 warn("OOB evaluation not possible: doing it in-bag."
                      " Performance evaluation is likely to be wrong"