This is an sklearn example, but AFAIK it should work also with XGB. 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?
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?
If you save it with joblib it should be automatically captured