配置模型

创建一个ReduceLROnPlateau回调函数,用于根据验证集上的损失减少学习率

reduce_lr = ReduceLROnPlateau(monitor='val_loss', factor=np.sqrt(0.1), patience=5)

创建一个Adam优化器,学习率为0.001

optimizer = Adam(learning_rate=0.001)

cnn_model.compile(optimizer=optimizer, loss=CategoricalCrossentropy(), metrics=['accuracy'])


原文地址: http://www.cveoy.top/t/topic/pggz 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录