Регистрация длинных тензорных значений в оценке тензорного потока - PullRequest
Новая
       6

Регистрация длинных тензорных значений в оценке тензорного потока

0 голосов
/ 21 декабря 2018

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

model = tf.estimator.DNNLinearCombinedClassifier(
        model_dir=model_dir,
        linear_feature_columns=wide_columns,
        dnn_feature_columns=deep_columns,
        dnn_hidden_units=hidden_units,
        config=run_config)
tensors_to_log = {"DenseOut": "dnn/logits/BiasAdd"}
logging_hook = tf.train.LoggingTensorHook(tensors=tensors_to_log, every_n_iter=1)
predictions = model.predict(train_input_fn, hooks=[logging_hook])

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

INFO:tensorflow:DenseOut = [[ 0.61572325 -0.44044942 -0.19232166 ...  0.04 0.605 0.15]]

Как мне указать тензор потока для записи полного вывода?

1 Ответ

0 голосов
/ 22 декабря 2018

Так интересно, я обнаружил, что решение было установить np.set_printoptions. 

import numpy as np

np.set_printoptions(threshold=np.nan)

Кажется, что tensorflow и numpy тесно интегрированы.

...