Я пытаюсь создать сверточную нейронную сеть для классификации набора данных CIFAR10.Цель состоит в том, чтобы изменить различные гиперпараметры, в моем случае оптимизатор, скорость обучения и размер пакета.Работает только изменение оптимизатора и скорости обучения, однако добавление размера пакета приводит к ошибке Value, указанной в заголовке.Любая помощь приветствуется: Ниже приведена часть моего кода.Я надеюсь, что это важная часть, но я все еще учусь, поэтому не уверен насчет этого.
Пока что я пытался изменить ключ в dict param_grid на «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)