Skip to content

Commit c26d478

Browse files
author
Florian Gardin
committed
improve doc for deduplication aglorithm
1 parent e80cb94 commit c26d478

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

skrules/skope_rules.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ class SkopeRules(BaseEstimator):
6666
than min_samples_split samples.
6767
6868
max_depth_duplication : integer or None, optional (default=3)
69-
The maximum depth of the decision tree for rule deduplication.
69+
The maximum depth of the decision tree for rule deduplication, if None then no deduplication occurs.
7070
7171
max_features : int, float, string or None, optional (default="auto")
7272
The number of features considered (by each decision tree) when looking
@@ -603,10 +603,12 @@ def _find_similar_rulesets(self, rules):
603603
Parameters
604604
----------
605605
rules : List, List of rules
606+
The rules that should be splitted in subsets of similar rules
606607
607608
Returns
608609
-------
609610
rules : List of list of rules
611+
The different set of rules. Each set should be homogeneous
610612
611613
"""
612614
def split_with_best_feature(rules, depth, exceptions=[]):
@@ -643,7 +645,7 @@ def split_with_best_feature(rules, depth, exceptions=[]):
643645

644646
def breadth_first_search(rules, leaves=None):
645647
if len(rules) == 0 or not isinstance(rules[0], list):
646-
if len(rules)>0:
648+
if len(rules) > 0:
647649
return leaves.append(rules)
648650
else:
649651
for rules_child in rules:

0 commit comments

Comments
 (0)