Skip to content

Commit 36f1afb

Browse files
authored
Update skope_rules.py (fix invalid mask indexing for OOB samples)
add ) in code
1 parent 26afac3 commit 36f1afb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

skrules/skope_rules.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -334,8 +334,9 @@ def fit(self, X, y, sample_weight=None):
334334
self.estimators_features_):
335335

336336
# Create mask for OOB samples
337-
mask = list(set(np.arange(len(samples))).difference(samples)
338-
# samples is numpy array. when you put ~. it returns array([-(value_1 + 1), ..., -(value_n + 1)])
337+
# mask = ~samples # samples is numpy array. when you put ~. it returns array([-(value_1 + 1), ..., -(value_n + 1)])
338+
mask = list(set(np.arange(len(samples))).difference(samples))
339+
339340

340341
if sum(mask) == 0:
341342
warn("OOB evaluation not possible: doing it in-bag."

0 commit comments

Comments
 (0)