我训练了我的第一个分类图像分类器,但是由于我是深度学习的新手,因此我没有注意到验证集正在发生什么。我正在做一个5折CV,在每个折中,我将数据集分为训练/验证/测试。
分割后,我在x_train中应用数据扩充:
datagen = ImageDataGenerator(
rotation_range=90,
horizontal_flip=True)
然后将我的模型与生成器拟合:
history = model.fit_generator(datagen.flow(x_train, y_train, batch_size=25),
validation_data=(x_validation,y_validation),
steps_per_epoch=math.ceil(len(x_train) / batch),
epochs=epochs)
参数:len(x_train)=31
和batch=1
因为我想要sgd批处理。
数据集中有76张图像,分别分为test = 15,Train = 31和Validation = 30。
运行5折后,它会显示以下图形:
发生了什么事?
谢谢!