Как напечатать значение метрики при обучении пользовательской модели кера с помощью оценщика? - PullRequest
Новая
       10

Как напечатать значение метрики при обучении пользовательской модели кера с помощью оценщика?

0 голосов
/ 19 сентября 2019

Я определил пользовательскую модель keras, используя Model subclass.И я хочу преобразовать эту модель в оценщик для распределенного обучения.Вот простые коды: 

    tf.logging.set_verbosity(tf.logging.INFO)

    model = EstimatorModel()

    model.compile(optimizer=tf.keras.optimizers.Adam(),
                  loss='categorical_crossentropy',
                  metrics=['accuracy'])

    estimator_model = tf.keras.estimator.model_to_estimator(
        keras_model=model, model_dir=FLAGS.model_path)

    estimator_model.train(input_fn=input_fn)

Код может работать нормально.Теперь вопрос в том, что оценщик печатает только loss, а не печатает metric.Я пытался добавить метрику к estimator_model, используя tf.estimator.add_metrics(estimator_model, my_auc), но он все еще не работает.Что я должен сделать, чтобы решить это?

...