import optuna
在原有代码上加,本质是,原本直接调用def train进行训练,加入之后,通过调用objective来调用def train进行训练,然后循环不同的参数进行训练。
参考:
- 工具介绍:使用 Optuna 进行超参数调优:https://jishuin.proginn.com/p/763bfbd71f56
- LightGBM+OPTUNA超参数自动调优教程(附代码框架):https://blog.csdn.net/yuxiaosmd/article/details/120249866