的 问题1: 强>
根据我的经验,在命名y,x变量时没有可辨别的限制。例如在 这个 一个人使用的内核 y_train, y_test 标签名称和 这里 一个人用 train_Y 。有一条规则,您应该给出显示变量内容的名称。
y_train, y_test
train_Y
的 问题2: 强>
我建议使用 validation_split 参数 ImageDataGenerator ( DOC )设置保留用于验证的图像部分。之后我会建议使用 subset 参数 flow_from_directory ( DOC ) 界定 training_generator 和 validation generator 变量。 (我想指出flow_from_directory返回生成器,而不是数据)。
validation_split
ImageDataGenerator
subset
flow_from_directory
training_generator
validation generator
所以你的代码看起来像:
data_generator = ImageDataGenerator( validation_split=0.2, rescale=1./255, shear_range=0.2, zoom_range=0.2, horizontal_flip=True, ) train_generator = data_generator.flow_from_directory( './images/train', target_size = (28, 28), batch_size = 86, class_mode = 'binary', color_mode='rgb', classes=None, subset="training" ) validation_generator = data_generator.flow_from_directory( './images/train', target_size = (28, 28), batch_size = 86, class_mode = 'binary', color_mode='rgb', classes=None, subset="validation" ) history=classifier.fit_generator( train_generator, steps_per_epoch = (8000 / 86), epochs = 2, validation_data = validation_generator, validation_steps = 8000/86, callbacks=[learning_rate_reduction] )