经过训练,我的模型一次输出所有的时代,并给出一个最终的性能指标。我怎样才能得到它,这样它就可以每10个时代打印一个带有性能指标的时代
model = baseline_model()
model.fit(X_train, y_train, epochs = 50)
y_pred = model.predict(X)
发布于 2019-08-09 13:46:35
使用回调
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)
参考文献:
https://stackoverflow.com/questions/57431103
复制相似问题