如何将训练和验证数据拟合在一起

时间:2021-04-28 05:31:47

标签: tensorflow model-fitting

history = model.fit(
    train_generator,
    validation_generator,
    steps_per_epoch= 2403//32,
    epochs= 5,
)

这是我目前正在使用的代码,但出现以下错误: ValueError: 'y' argument is not supported when using 'keras.utils.Sequence' as input. 我有 2403 张训练图像和 600 张验证图像,我使用的批大小为 32,请指导我如何做到这一点。

1 个答案:

答案 0 :(得分:0)

当您的输入是从生成器生成时,您需要按如下方式传递验证数据。

history = model.fit(
    train_generator,
    validation_data=validation_generator,
    steps_per_epoch= 2403//32,
    epochs= 5,
)
相关问题