【发布时间】:2019-08-09 13:27:06
【问题描述】:
训练后,我的模型一次输出所有时期,并给出一个最终性能指标。我怎样才能得到它,以便它每 10 个 epoch 打印一个带有性能指标的 epoch
model = baseline_model()
model.fit(X_train, y_train, epochs = 50)
y_pred = model.predict(X)
【问题讨论】:
标签: python tensorflow
训练后,我的模型一次输出所有时期,并给出一个最终性能指标。我怎样才能得到它,以便它每 10 个 epoch 打印一个带有性能指标的 epoch
model = baseline_model()
model.fit(X_train, y_train, epochs = 50)
y_pred = model.predict(X)
【问题讨论】:
标签: python tensorflow
使用回调
import keras
class Callback(keras.callbacks.Callback):
def __init__(self, acc_threshold):
super(Callback, self).__init__()
def on_epoch_end(self, epoch, logs={}):
if epoch % 10 == 0:
performance_indicator()
例子:
model = baseline_model()
c = Callback(0.05)
model.fit(X_train, y_train, epochs=50, callbacks=[c])
y_pred = model.predict(X)
参考资料:
【讨论】: