from sklearn.model_selection import train_test_split from sklearn.metrics import accuracy_score from sklearn.svm import SVC import optuna from sklearn.datasets import load_iris import pandas as pd x = pd.DataFrame(load_iris().data, columns=load_iris().feature_names) y = load_iris().target ''' optuna根据需要,支持多种不同的搜索方式(含有low和high值的都是左闭右开): 对待搜索的超参数指定一个取值空间,通过trial参数来实现: trial.suggest_int(name,value1,value2):为超参数指定整型取值 trial.suggest_float(name,value1,value2):为超参数指定浮点型取值 trial.suggest_categorical(name,value1,value2):为超参数指定类型取值 trial.suggest_uniform(name,value1,value2):为超参数指定取值空间,且对每个值进行相同采样 trial.suggest_loguniform(name,value1,value2):为超参数指定取值空间,且在对数域对每个值进行相同采样 trial.suggest_discrete_uniform(name,value1,value2):为超参数指定取值空间,且对每个值以相同可能采样 ''' def objective(trial, x, y): X_train, X_test, y_train, y_test = train_test_split(x, y, t
模型调参(机器学习)
最新推荐文章于 2024-04-21 10:25:46 发布