上一节中我们搭建了CNN网络,这一节我们开始使用:
第一:调用CNN
model = CNN()
第二:设置损失函数,优化器和准确率(损失函数和优化器的选择不同情况不一样选择)
model.compile(loss=tf.keras.losses.categorical_crossentropy,# 分类的交叉熵损失 #loss传值有三种方式
optimizer=tf.keras.optimizers.Adam(),# 随机梯度下降
metrics=['accuracy'])
第三:模型参数设置:
model.fit(train_generator,
steps_per_epoch=None,
epochs=20,
validation_data=valid_generator,
validation_steps=None,
validation_freq=1,
class_weight=None,
max_queue_size=10,
workers=1,
use_multiprocessing=False,
shuffle=True,
initial_epoch=0)
其中:
steps_per_epoch=None, #不设