【发布时间】:2018-12-22 20:11:59
【问题描述】:
我正在使用 Tensorflow 的 Dataset 函数和 TFRecord 在 Keras 中构建模型。我用 Keras 训练模型成功了,但是问题出在 val_loss 上。它根本没有显示在 Keras 的进度条中。
if __name__ == '__main__':
x_train,y_train = input_fn('train_whale_without07.tfrecords')
x_test,y_test = input_fn('test_whale_without07.tfrecords')
img_input = layers.Input(tensor = x_train)
model = CNN(img_input)
model.compile(optimizer=Adam(lr=0.001), loss='categorical_crossentropy',
metrics=[categorical_crossentropy, categorical_accuracy],
target_tensors=[y_train])
model.fit(steps_per_epoch=3000, epochs=EPOCHS, batch_size=None, verbose=1, validation_data = ([x_test,y_test]))
model.save('my_model_keras.h5')
结果是这样的
纪元 1/15 1/3000 [.......................] - ETA:00:05:12 - 损失:8.1786 - categorical_crossentropy: 8.1786 - 分类精度:0.0000e+00
有谁知道如何添加 val_loss 吗?
【问题讨论】:
标签: tensorflow keras