我有一个Keras顺序模型,正在使用:
model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])
使用Keras fit()
函数训练模型时,我可以看到打印的训练精度。
我需要使用Estimator API训练模型,而我正在使用model_to_estimator
将模型转换为estimator。然后,我使用train_and_evaluate()
训练模型。
但是我在Tensorboard中看不到精度图。仅有一个精度值(来自评估),因此图形只是一个点。
我需要的是来自训练的精度图,如下所示: https://www.tensorflow.org/guide/custom_estimators#tensorboard
我检查了这些示例,发现所有这些示例都是使用Estimator API构建模型并使用以下代码定义摘要标量的。
# Compute evaluation metrics.
accuracy = tf.metrics.accuracy(labels=labels,
predictions=predicted_classes,
name='acc_op')
metrics = {'accuracy': accuracy}
tf.summary.scalar('accuracy', accuracy[1])
有人知道如何与从Keras转换的模型一起使用吗?
我正在使用Tensorflow版本r1.10。