def _get_node(self, data, target, level=1):
print('level=', level)
if level > self.split_count:
node = TreeNode(level=level)
else:
node, meta = self._calc_split(data, target)
node.level = level
if not node.is_leaf():
node.lo_branch = self._get_node(meta.get('lo_data'),
target,
level=++level)
node.lo_branch.w = meta.get('lo_w')
node.hi_branch = self._get_node(meta.get('hi_data'),
target,
level=++level)
node.hi_branch.w = meta.get('hi_w')
print('factor=', node.factor, 'value=', node.value)
return node
def test_node_empty(self):
n = TreeNode()
self.assertEqual(True, n.is_leaf())