google / yggdrasil-decision-forests

A library to train, evaluate, interpret, and productionize decision forest models such as Random Forest and Gradient Boosted Decision Trees.
https://ydf.readthedocs.io/
Apache License 2.0
498 stars 53 forks source link

Go module only supports binary classification #54

Open jimidle opened 1 year ago

jimidle commented 1 year ago

This is documented, but is quite limiting. How much effort would it take to support this? I might be willing to contribute if there are no resources available to do this.

rstz commented 1 year ago

Hi, see my answer in the discussions thread, the core team does not have the bandwidth to work on this in the near future, but we're happy to review and discuss PRs.