使用model_to_estimator的训练精度图

时间:2018-09-20 15:29:02

标签: python tensorflow keras tensorflow-estimator

我有一个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。

0 个答案:

没有答案