Как я могу получить feed_dict, используя ловушку, когда я запускаю estimator.predict - PullRequest
Новая
       4

Как я могу получить feed_dict, используя ловушку, когда я запускаю estimator.predict

0 голосов
/ 12 июня 2019

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

Я пробовал код ниже, но feed_dict выводит None, я вызываю предикат, используя dataset объект. 

class InsertPredictResultHook(tf.estimator.SessionRunHook):
    def after_run(self, run_context, run_values):
        print(run_context.original_args.fetches)
        print(run_context.original_args.feed_dict)
        print(run_values)

predictions = classifier.predict(input_fn=mydataset,
    hooks = [InsertPredictResultHook()])

Как я могу получить входные тензоры вместе с прогнозируемым результатом, чтобы я мог сохранить эти данные для моих пользователей?

Я использую tenorflow 2.0 alpha0

...