Получить значение тензора (Numpy) после обучения tf.Estimator - PullRequest
0 голосов
/ 01 июня 2018

Я обучил простой автоэнкодер, используя tf.Estimator.Во время обучения обновляется конкретный тензор W, где W - матрица.После обучения я хотел бы получить W и прочитать его значения, используя numpy.

. Это простая задача, если бы я не использовал tf.Estimator, я бы позвонил .eval() и пропустил мой сеанс.Однако Estimator является высокоуровневым API, и инициализация и использование сеанса выполняются изнутри.

Я также пытался использовать Estimator.predict и EstimatorSpec для возврата W, но это не так.не похоже на работу.Я получаю следующую ошибку:

TypeError: Список тензоров, когда ожидается один тензор.

Можно ли напрямую получить значение тензора numpy после тренировкис tf.Estimator.Если да, то как?

1 Ответ

0 голосов
/ 02 июня 2018

Предполагая, что W хранится как переменная в вашей модели, вы можете использовать метод get_variable_value объекта Estimator.Смотри здесь .

...