Keras预处理层是否适用于验证集?



我正在阅读关于Keras的数据增强文章,他们允许人们将预处理层作为模型的一部分:

model = tf.keras.Sequential([
resize_and_rescale,
data_augmentation,
layers.Conv2D(16, 3, padding="same", activation="relu"),
layers.MaxPooling2D(),
# Rest of your model
])

我想知道是否一个或两个resize_and_rescale和data_augmentation层也应用于训练期间的验证集?

这取决于您使用的是哪种类型的增强。例如,如果您使用调整大小层或重新缩放层,它们甚至在推理模式中应用,也就是说,它们将应用于model.fit中的验证数据。对于其他增强层,如RandomFlip层,文档说明:

During inference time, the output will be identical to input.

所以你必须查找你正在使用的图层类型的信息。文档在这里。从我能收集到的,我认为只有调整大小和重新缩放图层在推理模式下保持活跃。

最新更新