Skip to content

Commit 75d71ce

Browse files
Merge pull request #31 from floriangardin/master
add sort of rules
2 parents 59c03c5 + 59531c5 commit 75d71ce

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

skrules/skope_rules.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -377,6 +377,7 @@ def fit(self, X, y, sample_weight=None):
377377
# Deduplicate the rule using semantic tree
378378
if self.max_depth_duplication is not None:
379379
self.rules_ = self.deduplicate(self.rules_)
380+
self.rules_ = sorted(self.rules_, key=lambda x: - self.f1_score(x))
380381
return self
381382

382383
def predict(self, X):

0 commit comments

Comments
 (0)