gpt4 book ai didi

python - 我无法在 gridsearch 中添加优化器参数

转载 作者:太空宇宙 更新时间:2023-11-04 09:35:53 25 4
gpt4 key购买 nike

from keras.wrappers.scikit_learn import KerasClassifier
from sklearn.model_selection import GridSearchCV
def build_classifier():
classifier = Sequential()
classifier.add(Dense(units = 6 , init='uniform' , activation= 'relu'))
classifier.add(Dense(units = 6 , init='uniform' , activation= 'relu'))
classifier.add(Dense(units = 1 , init='uniform' , activation= 'sigmoid'))
classifier.compile(optimizer='adam' , loss = 'binary_crossentropy' ,
metrics=['accuracy'])
return classifier
KC = KerasClassifier(build_fn=build_classifier)
parameters = {'batch_size' : [25,32],
'epochs' : [100,500],
'optimizer':['adam','rmsprop']}
grid_search = GridSearchCV(estimator=KC ,
param_grid=parameters,scoring='accuracy',cv=10)
grid_search.fit(X_train,y_train)

我想用不同的优化器测试模型。但我似乎无法在网格搜索中添加优化器。每当我运行该程序时,它都会显示有关拟合训练集的错误。

ValueError:优化器不是合法参数

最佳答案

keras for scikit-learn 的文档说:

sk_params takes both model parameters and fitting parameters. Legalmodel parameters are the arguments of build_fn. Note that like allother estimators in scikit-learn, build_fn should provide defaultvalues for its arguments, so that you could create the estimatorwithout passing any values to sk_params.

GridSearchCV 将在 KerasClassifier 上调用 get_params() 以根据您的代码获取可以传递给它的有效参数列表:

KC = KerasClassifier(build_fn=build_classifier)

将为空(因为您没有在 build_classifier 中指定任何参数)。

将其更改为:

# Used a parameter to specify the optimizer
def build_classifier(optimizer = 'adam'):
...
classifier.compile(optimizer=optimizer , loss = 'binary_crossentropy' ,
metrics=['accuracy'])
...
return classifier

之后它应该可以工作了。

关于python - 我无法在 gridsearch 中添加优化器参数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53806892/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com