Я обучил простой автоэнкодер, используя tf.Estimator
.Во время обучения обновляется конкретный тензор W
, где W
- матрица.После обучения я хотел бы получить W
и прочитать его значения, используя numpy
.
. Это простая задача, если бы я не использовал tf.Estimator
, я бы позвонил .eval()
и пропустил мой сеанс.Однако Estimator
является высокоуровневым API, и инициализация и использование сеанса выполняются изнутри.
Я также пытался использовать Estimator.predict
и EstimatorSpec
для возврата W
, но это не так.не похоже на работу.Я получаю следующую ошибку:
TypeError: Список тензоров, когда ожидается один тензор.
Можно ли напрямую получить значение тензора numpy
после тренировкис tf.Estimator
.Если да, то как?