График точности обучения с model_to_estimator - PullRequest
Новая
       6

График точности обучения с model_to_estimator

0 голосов
/ 20 сентября 2018

У меня есть последовательная модель Keras, и я использую: 

model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])

Я вижу напечатанную точность обучения, когда использую функцию Keras fit() для обучения модели.

Мне нужно использовать Estimator API для обучения модели, и я использую model_to_estimator для преобразования модели в оценщик.Затем я использую train_and_evaluate() для обучения модели.

Однако я не вижу график точности в Tensorboard.Есть только одно значение точности (из оценки), поэтому график - это просто точка.

Мне нужен график точности обучения, подобный показанному здесь: https://www.tensorflow.org/guide/custom_estimators#tensorboard

Я проверил примеры, и все, что я смог найти, это те, где они используют API Estimator дляпостроить модель и использовать следующий код для определения итогового скаляра. 

# 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.

...