pyensemble/wildwood

Bug while predict with a single node tree

Opened this issue · 0 comments

In an extreme case, if we construct a tree whose root is a leaf, then this will raise a bug: in particular, tree_classifier_predict_proba cannot handle the case while node[0] has no child (as well as tree_regressor_predict), in _tree.py