核心设置
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')