From 4485f76727be2e7271a25ca84db9ff96fd69a9d7 Mon Sep 17 00:00:00 2001 From: Mike Vink Date: Fri, 18 Sep 2020 09:09:59 +0200 Subject: Minor: improved stop condition for tree.predict --- assignment1.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'assignment1.py') diff --git a/assignment1.py b/assignment1.py index 00432ec..ac3b7ec 100644 --- a/assignment1.py +++ b/assignment1.py @@ -51,7 +51,7 @@ class Node: def is_leaf_node(self, node_classes): """ - is_leaf_node is used to change the col attribute to zero to indicate a + is_leaf_node is used to change the col attribute to None to indicate a leaf node """ self.col = None @@ -84,7 +84,7 @@ class Tree: # De index van de row van x die we in de boom willen droppen drop = 0 - while not set(nodes).issubset({0, 1}): + while not all(pred_class in {0,1} for pred_class in nodes): # Als de col None is dan is het een leaf node, dus dan is de row # van x hier beland if nodes[drop].col is None: -- cgit v1.2.3