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