【发布时间】:2022-12-14 22:39:36
【问题描述】:
我正在尝试使用 Optuna 调整一个额外的树分类器。
我在所有试验中都收到了这条消息:
[W 2022-02-10 12:13:12,501] 试验 2 失败,因为值为 None 无法漂浮。
下面是我的代码。它发生在我所有的试验中。谁能告诉我我做错了什么?
def objective(trial, X, y): param = { 'verbose': trial.suggest_categorical('verbosity', [1]), 'random_state': trial.suggest_categorical('random_state', [RS]), 'n_estimators': trial.suggest_int('n_estimators', 100, 150), 'n_jobs': trial.suggest_categorical('n_jobs', [-1]), } X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, shuffle=True, random_state=RS) clf = ExtraTreesClassifier(**param) clf.fit(X_train, y_train) y_pred = clf.predict(X_test) acc = accuracy_score(y_pred, y_test) print(f"Model Accuracy: {round(acc, 6)}") print(f"Model Parameters: {param}") print('='*50) return` study = optuna.create_study( direction='maximize', sampler=optuna.samplers.TPESampler(), pruner=optuna.pruners.HyperbandPruner(), study_name='ExtraTrees-Hyperparameter-Tuning') func = lambda trial: objective(trial, X, y) %%time study.optimize( func, n_trials=100, timeout=60, gc_after_trial=True )
【问题讨论】:
-
你解决了吗?我有同样的问题
标签: performance hyperparameters optuna