Merge b210973 into f6ab4a7

svaante · Jan 28, 2019 · 531b363 · 531b363
2 parents f6ab4a7 + b210973
commit 531b363
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/id3/tree.py b/id3/tree.py
@@ -131,7 +131,7 @@ def _prune(self, node, tree):
                                                  - node.predicts[max_class],
                                                  n_predicts)
 
-                if node_error_rate < children_error_rate:
+                if node_error_rate <= children_error_rate:
                     node.is_feature = False
                     node.value = max_class
                     node.n_correct_predicts = node.predicts[max_class]

diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
     sys.exit(1)
 
 setup(name='decision-tree-id3',
-      version='0.1.2',
+      version='0.1.3',
       description='A scikit-learn compatible package for id3 decision tree',
       author='Daniel Pettersson, Otto Nordander',
       packages=find_packages(),