卷积神经网络的交叉验证

数据挖掘 深度学习 喀拉斯 张量流 计算机视觉 卷积神经网络
2022-02-16 01:54:54

我正在使用 Keras 创建 CNN 模型,并且我会使用K-fold 交叉验证来训练数据集。

数据集包含图像,我正在使用flow_from_directory函数。

你知道如何在 Keras 中使用K-fold 交叉验证来创建 CNN 模型吗?

1个回答

使用sklearn 进行交叉验证是很常见的。

最知名的方法是从sklearn.model_selection导入的KFoldcross_val_score

然后,您可以使用tf.keras.wrappers.scikit_learn.KerasClassifier来实现带有Keras模型的 scikit-learn 分类器