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