【发布时间】:2020-02-25 15:48:40
【问题描述】:
我是深度学习的新手,正在做一些分类问题。
我在回调列表中使用EarlyStopping 和ModelCheckpoint,但是当训练开始时,模型检查点的基线是负无穷大并覆盖“best_model.h5”。
但是,“best_model.h5”已经存储了我最后一个最好的模型。我想将ModelCheckpoint 的基线设置为我在数据上的最后一个最佳模型的性能。
谁能帮帮我?
es = EarlyStopping(monitor='val_accuracy', mode='max', verbose=1, patience=3)
mc = ModelCheckpoint('best_model.h5', monitor='val_accuracy', mode='max', save_best_only=True, verbose=1)
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(x_train, y_train, validation_data=(x_valid, y_valid), batch_size=400,\
epochs=20, callbacks=[es, mc])
【问题讨论】:
标签: keras deep-learning