If you save it with joblib it should be automatically captured

https://github.com/allegroai/clearml/blob/master/examples/frameworks/scikit-learn/sklearn_joblib_example.py

This is an sklearn example, but AFAIK it should work also with XGB. Makes sense?

I'll try, thanks!

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 call
task.set_model_label_enumeration({'label':0,'another_label':1})The models will inherit this enumeration 🙂
Makes sense?

yep, will try, thanks!