LG: num_leaves=1 support

dmitryikh / leaves

pure Go implementation of prediction part for GBRT (Gradient Boosting Regression Trees) models from popular frameworks

MIT License

419 stars 72 forks source link

LG: num_leaves=1 support #39

Closed dmitryikh closed 5 years ago

dmitryikh commented 5 years ago

LightGBM tree can be like next:

Tree=2236
num_leaves=1
num_cat=0
split_feature=
split_gain=
threshold=
decision_type=
left_child=
right_child=
leaf_value=0
leaf_count=
internal_value=
internal_count=
shrinkage=1.03754e-322

But leaves treats num_leaves < 2 as input error.

TODO:

support a tree with only one leaf
add test for this case