介绍
Optuna库是机器学习中提供自动化超参数优化的库,它和很多知名的机器学习库
(如sklearn,keras,tensorflow,xgboost,lightgbm,pytorch等)是兼容的。
dashboard是其中一个提供结果画图展示和可交互操作的功能,在当前的版本中已经被独立出来成为一个模块,下面将用一个简单的示例来展现它的效果。
示例
在使用之前,如果没有安装,可以使用以下命令进行安装。
conda install -c conda-forge optuna
或者
pip install optuna
安装好optuna库后,再安装optuna-dashboard。
conda install -c conda-forge optuna-dashboard
或者
pip install optuna-dashboard
安装完毕后先用optuna创建好我们要用的study,将它保存在数据库中,然后用dashboard导入即可,具体如下:
import optuna
import sklearn
def objective(trial):
Classifier_name = trial.suggest_categorical('classifier', ['GradientBoosting', 'RandomForest'])
if Classifier_nam