python - Keras:我可以使用 model.predict 但不使用 model.predict_generator 来预测是否使用 model.fit_generator 训练模型

Question

我是keras的新手。现在，我将使用我使用 model.fit_generator 训练的模型来预测测试图像组。我可以使用 model.predict 吗？不确定如何使用model.predict_generator。并且文献表明这两者的结果不同，哪一个更好？

非常感谢!

datagen = ImageDataGenerator(
    enter code here`rotation_range=
    ...
    zoom_range = 0.05) 


model = Sequential()
model.add 
...


model.fit_generator(datagen.flow(train_x, train_y, batch_size=batch_size), 
    steps_per_epoch=train_x.shape[0] // batch_size, 
    epochs=epochs, validation_data=(test_x, test_y))


from sklearn.metrics import roc_auc_score
test_y_hat = model.predict(test_x)
roc_auc_score(test_y, test_y_hat)

Answer 1

是的，你可以做到看看code对于 fit_generator()，您可以看到模型不记住它是如何训练的。事实上你可以使用所有这些API(fit、fit_generator、predict、predict_generator) 以任意顺序在 keras 模型上执行。