Downloads · 30 days
0
text_concat_xgb_v1.0
InfiniFlow/text_concat_xgb_v1.0
text_concat_xgb_v1.0 is a machine learning model from InfiniFlow. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Access
Public
Updated Dec 14, 2023
Repo size
5.9 MB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.model5.9 MB · 100%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Dec 14, 2023
- Updated
- Dec 14, 2023
- SHA
- 722ed09a
Model card
From the Hugging Face model README
Model Loading
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]
- License
- apache-2.0
- Created
- Dec 14, 2023
- Updated
- Dec 14, 2023