上一节中我们搭建了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, #不设置参数steps_per_epoch(是通过把训练图像的数量除以批次大小得出的)
epochs=20, #训练20轮
validation_data=valid_generator, #validation_data用来在每个ep

本节继续上一节的CNN网络搭建,重点在于如何调用CNN网络,设置损失函数、优化器和准确率,并详细阐述了模型参数配置,包括epochs、validation_data、validation_steps、class_weight等关键参数,旨在实现有效的训练过程并应对不平衡数据集。

4405





