CNN网络的使用

上一节中我们搭建了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, #不设