xgboost json model fails to load

[LLautenbacher]

Hi,

I have a xgboost model in JSON format when I try to load it with treelite version >4.0 (treelite.frontend.load_xgboost_model) I get the error Node with key 1 is orphaned -- it cannot be reached from the root node. With version 3.9 (treelite.Model.load) it worked without issue. Do you have an idea what is causing this, or how I can convert the json into a format that works with treelite >4.0?

You can download the model file from Zenodo: https://zenodo.org/records/7576604

Thank you for your help in advance!

[hcho3]

Can you also share the script you used for training the XGBoost model? I'd like to know the list of hyperparameters.

[LLautenbacher]

I didn't train the model myself. But this is probably the script that was used to train it and this should be the hyperparameters selected during tuning.

Taken from the supplement of this paper.

[hcho3]

@LLautenbacher The bug should be fixed in the upcoming Treelite release. Thanks!