sklearn.model_selection.GridSearchCV 和 scikeras.wrappers.KerasClassifier 之间的问题

问题描述 投票:0回答:1

我的sklearn版本是1.4.2,我的版本是0.13.0,当我想在分类中使用gridsreach时,我得到了这个错误:

ValueError:估计器 KerasClassifier 的参数激活无效。 通过在 KerasClassifier 构造函数中设置此参数可能可以解决此问题:

KerasClassifier(activation=relu)
使用
estimator.get_params().keys()

检查可用参数列表
keras_wr_class = KerasClassifier(build_fn=AnnModel)
grid_search = GridSearchCV(estimator=keras_wr_class, param_grid=param_grid, scoring='accuracy', cv=5)
grid_search.fit(x_train_s,y_train)
keras classification scikeras
1个回答
0
投票

要使用 GridSearchCV 微调超参数,请首先安装

scikeras library
。预处理数据并定义模型。接下来,定义一个 包含
hyperparameter
名称及其潜在值的字典。裹 KerasClassifier 类中的模型。最后传递包装好的模型 和超参数字典到 GridSearchCV 并启动训练 用fit方法进行处理。我已经实现了 GridSearchCv ,请参阅 提供了要点

© www.soinside.com 2019 - 2024. All rights reserved.