Hi! So I looked at the example code and you don't have to use joblib 😄
If you do:model.save_model('xgb.02.model')
It will work 🙂 Sorry for the confusion!
To upload labels, after task.init calltask.set_model_label_enumeration({'label':0,'another_label':1})
The models will inherit this enumeration 🙂
Makes sense?
Ok it works, but I don't see the labels in the model output, is there a way for me to use OutputModel to update those labels?
This is an sklearn example, but AFAIK it should work also with XGB. Makes sense?
If you save it with joblib it should be automatically captured