search

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

support v3 model encoding #66

Open adamgreenhall opened 4 years ago

adamgreenhall commented 4 years ago

lightgbm changed their model encoding to v3 in v2.3.0 to support weights in the model. Would like to see leaves support this new format.

dmitryikh commented 4 years ago

Thanks for the issue.

I'm going to fix this.

arnwas commented 4 years ago

I left a quick workaround(?) in a PR for that. Seems to work well for me.