python - 将图像生成器与 GridSearchCV 配合使用

Question

我ImageDataGenerator用keras. 所以，我有train_generator和。但是函数接受两个变量。例子：。但我没有这些数据。我如何将 fit 生成器与网格搜索一起使用？我的代码：valid_generatortest_generatorfitGridSearchCVgrid.fit(X,y)

def create_model(learn_rate = 0.001,momentum = 0):
    model.add(
    .
    .
    .

    return model


model = KerasClassifier(build_fn = create_model, verbose = 0)

grid = GridSearchCV(estimator = model, param_grid = param_grid, n_jobs = -1, cv = 10)


grid_result = grid.fit(??)

标签： pythonmachine-learningkerasscikit-learngrid-search