【发布时间】:2019-05-17 07:52:37
【问题描述】:
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 和 scikit-learn 版本是什么?
标签: python keras scikit-learn deep-learning neural-network