torch_frame.gbdt.LightGBM

class LightGBM(task_type: TaskType, num_classes: int | None = None, metric: Metric | None = None)[source]

基础类：GBDT

使用Optuna进行超参数调优的LightGBM实现。

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

objective(trial: Any, train_data: Any, eval_data: Any, num_boost_round: int) → float[来源]

需要优化的目标函数。

Parameters:

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

  • train_data (lightgbm.Dataset) – 训练数据。

  • eval_data (lightgbm.Dataset) – 验证数据。

  • num_boost_round (int) – 提升轮数。

Returns:

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

Return type:

float