在官方文件中张量流keras https://www.tensorflow.org/api_docs/python/tf/keras/Model,
validation_data 可以是:Numpy 数组或张量的元组 (x_val, y_val)
Numpy 数组的元组 (x_val, y_val, val_sample_weights)
对于前两种情况,必须提供batch_size。对于最后一种情况,可以提供validation_steps。
它没有提到生成器是否可以充当validation_data。所以我想知道validation_data是否可以是一个数据生成器?像以下代码 https://github.com/filipetrocadoferreira/end2endlobesegmentation/blob/master/train.py:
net.fit_generator(train_it.generator(), epoch_iterations * batch_size, nb_epoch=nb_epoch, verbose=1,
validation_data=val_it.generator(), nb_val_samples=3,
callbacks=[checker, tb, stopper, saver])
更新:
在官方文件中keras https://keras.io/models/sequential/,内容相同,但多了一句:
考虑到
对于前两种情况,必须提供batch_size。对于最后一种情况,可以提供validation_steps。
我觉得应该有3种情况。 Keras 的文档是正确的。所以我会在tensorflow.keras中发布一个问题来更新文档。