torch_frame.gbdt.XGBoost

class XGBoost(task_type: TaskType, num_classes: int | None = None, metric: Metric | None = None)

基础类：GBDT

使用Optuna进行超参数调优的XGBoost模型实现。

此实现扩展了GBDT，旨在通过优化给定的目标函数来找到最佳超参数。

objective(trial: Any, dtrain: Any, dvalid: Any, num_boost_round: int, early_stopping_rounds: int) → float

需要优化的目标函数。

Parameters:

• 试验 (optuna.trial.Trial) – Optuna 试验对象。

• dtrain (xgboost.DMatrix) – 训练数据。

• dvalid (xgboost.DMatrix) – 验证数据。

• num_boost_round (int) – 提升轮数。

• early_stopping_rounds (int) – 提前停止的轮数。

Returns:

最佳目标值。回归任务的均方根误差和分类任务的准确率。

Return type:

float