Keras 值\u错误：批大小不是合法参数

我正在尝试创建一个卷积神经网络来对CIFAR10数据集进行分类。目标是改变各种超参数，在我的例子中是优化器、学习率和批量大小。只有改变优化器和学习率才能很好地工作，但是添加批量大小会抛出标题中引用的值错误。感谢您的帮助： 下面是我代码的一部分。我希望这是相关的部分，但我仍在学习，所以对此不确定

到目前为止，我试图将dict param_网格中的键更改为“batchsize”，因为我认为这可能是语法错误。我还检查了互联网上允许哪些参数进入函数GridSearchCV param_grid，可惜我没有找到它

model = KerasClassifier(
build_fn=create_model,
epochs=epochs,
verbose=1)

optimizer_candidates = [Adam, RMSprop]
lr_candidates = [1e-3, 5e-3, 1e-4]
batch_size = [128, 64]


param_grid = {
    "optimizer": optimizer_candidates,
    "lr": lr_candidates,
    "batch-size": batch_size}

grid = GridSearchCV(
    estimator=model,
    param_grid=param_grid,
    n_jobs=1,
    verbose=1,
    cv=3)

grid_result = grid.fit(x_train, y_train)

---

它可能应该是“批次大小”而不是“批次大小”。

您的

参数网格中批次大小的名称应该是
“批次大小”
，而不是
“批次大小”
谢谢，完全忽略了谢谢，完全忽略了这一点