修改深度模型学习率,亲测有效。
导入包:
from keras.callbacks import ReduceLROnPlateau
学习率设置:
reduce_lr = ReduceLROnPlateau(monitor='loss', patience=10, mode='auto')
history = model.fit(train_X, train_y, epochs=50, batch_size=72, validation_data=(test_X, test_y), verbose=2, shuffle=False, callbacks=[reduce_lr])
monitor可以选择 'val_loss' 或者 'loss'