import xgboost as xgb
import torch
model = xgb.Booster()
if torch.cuda.is_available():
model.set_param({"device": "cuda"})
model.load_model('InfiniFlow/text_concat_xgb_v1.0')
Prediction
model.predict(xgb.DMatrix([feature]))[0]
Downloads last month
-
Downloads are not tracked for this model.
How to track
Unable to determine this model's library. Check the
docs
.
