Export Model

[Eyal-Elbaz]

Hi there,
I have a XGBoost model like this:

var trainer = new XGBoostRegressionTrainer(XGBoostRegressionTrainer.RegressionType.LINEAR, 50,50,true);
var model = trainer.train(trainData);

and I want to save him and load him like this but I didn't manage

[Craigacp]

Tribuo doesn't expose the XGBoost4J Booster class that you could use to save it following the XGBoost4J instructions. You can serialise Tribuo models directly via an ObjectOutputStream and deserialise them using an ObjectInputStream. Unfortunately in v4.0.1 regression model serialization is broken (see #67), but that's fixed in the main branch and v4.0.2 which is coming very soon.

If you want to train the model in Tribuo and load it in another system then we're considering ONNX export support. We can look at exposing the internal Boosters too.

[Eyal-Elbaz]

how can I export using ONNX?

[Craigacp]

You can't yet. The onnx model converters are all written in python, so we need to figure out how to build equivalent operations in Java using the onnx c API.

However PR #79 does expose the internal Booster objects so you can save those using the XGBoost4J API.

[Eyal-Elbaz]

Ok, thanks, I will try.