| 2 | from collections import Counter |
| 3 | |
| 4 | class Node: |
| 5 | def __init__(self, feature=None, threshold=None, left=None, right=None,*,value=None): |
| 6 | self.feature = feature |
| 7 | self.threshold = threshold |
| 8 | self.left = left |
| 9 | self.right = right |
| 10 | self.value = value |
| 11 | |
| 12 | def is_leaf_node(self): |
| 13 | return self.value is not None |
| 14 | |
| 15 | |
| 16 | class DecisionTree: |