Support XGBoost 1.6 - Githubissues

dmlc / treelite

Universal model exchange and serialization format for decision tree forests

https://treelite.readthedocs.io/en/latest/

Apache License 2.0

729 stars 98 forks source link

Closed hcho3 closed 10 months ago

hcho3 commented 2 years ago

hcho3 commented 2 years ago

@trivialfis Anything else we should add to this list?

trivialfis commented 2 years ago

The num_parallel_tree is now a model parameter.

trivialfis commented 2 years ago

I don't think it's necessary to support ubjson in treelite unless the current json loading is not efficient enough.

hcho3 commented 2 years ago

Got it, I'll postpone UBJSON support for the time being.

hcho3 commented 2 years ago

~It turns out Treelite doesn't support multi-class classifiers with num_parallel_tree > 1.~ Fixed in #383

hcho3 commented 2 years ago

The current version of Treelite throws a clear error when a multi-target model is given:

num_target must be 1; Treelite doesn't support multi-target regressor yet

I will create a new version of Treelite now. The remaining TODO items will be tackled in a future version.

hcho3 commented 10 months ago

Closing. Treelite 4.0 can handle multi-output models just fine.