catboost推理开GPU加速

核心设置

model.predict(feature, task_type=‘GPU’)

代码参考

# 训练配置
params = {
	"catboost": {
	    "n_estimators": 7000,
	    "learning_rate": 0.03,
	    "eval_metric": "AUC",
	    "loss_function": "RMSE",
	    "random_seed": 2022,
	    "metric_period": 500,
	    "od_wait": 500,
	    "task_type": "GPU",  ## 这里开GPU加速训练
	    "depth": 8
	    }
}

# 推理设置
model = CatBoostRegressor(**params["catboost"])
model = model.fit(feature, label)
model.predict(feature, task_type='GPU')