首页 > 解决方案 > 如果我只是将数据集提供给我的 Keras 模型,我是否在使用验证数据集?

问题描述

所以我的问题是关于fit函数以及如果我只是向它提供数据集会发生什么,如下所示:

trainDataset = train_parsed_image_dataset.map(buildInputsLabels).shuffle(2000).batch(batches)
history = model.fit(trainDataset, epochs=epochs)

该模型使用什么作为验证数据集?它不使用任何验证数据集吗?我知道我可以明确地告诉我要用于验证的内容,但在这种情况下,我只提供数据集会发生什么?

提前致谢!

标签: pythonkeras

解决方案


如果您不定义validation_split> 0 (默认为 0),它将不会使用任何验证数据。它将使用相同的训练数据来评估loss& accuracy。这可能会导致过拟合。

参考:https ://github.com/keras-team/keras/issues/2279


推荐阅读