Views
No views yet
1import xgboost as xgb 2import torch 3 4model = xgb.Booster() 5if torch.cuda.is_available(): 6model.set_param({"device": "cuda"}) 7model.load_model('InfiniFlow/text_concat_xgb_v1.0')
model.predict(xgb.DMatrix([feature]))[0]