使用Tensorflow可视化多个嵌入

时间:2017-07-10 20:24:01

标签: tensorflow visualization embedding

我想基于多个张量变量来显示我的数据,即基于不同的嵌入变量。换句话说,我需要做的是:

我需要将100维矢量(图像特征/嵌入)存储到5个不同的变量中。然后我需要根据5个不同的变量来显示我的数据。也就是说,我需要根据前20个功能可视化我的数据,并基于第二个20个功能等等...

当我在https://www.tensorflow.org/get_started/embedding_viz上查看嵌入式可视化教程时,他们说我们可以添加多个嵌入。这就是我要找的。

如何在tensorflow中执行此操作?

非常感谢任何帮助!!

1 个答案:

答案 0 :(得分:0)

所以它不起作用的原因是因为我试图将100维嵌入分成100个不同的变量。那没用。因此,当我将嵌入分为5个不同的部分时,即将它们分成5个不同的变量,就可以了。以下是我的代码:

import numpy as np
import tensorflow as tf
from tensorflow.contrib.tensorboard.plugins import projector

LOG_DIR = \
    'C:/Users/user/PycharmProjects/VariationalAutoEncoder/' \
    'Tensorflow-DeconvNet-Segmentation/Embeddings/features_images.ckpt'

feature_vectors = np.loadtxt('features.txt')
feature_vectors = feature_vectors[:5329]

print("feature_vectors_shape:",feature_vectors.shape)

sub_features = []
for i in range(20):
    features = tf.Variable(feature_vectors[:, 5 * i: 5 * (i + 1)], name=('features' + str(i)))
    sub_features.append(features)

with tf.Session() as sess:

    saver = tf.train.Saver()
    sess.run(tf.global_variables_initializer())
    saver.save(sess, LOG_DIR)

    config = projector.ProjectorConfig()
    for i in range(20):
        embedding = config.embeddings.add()
        embedding.tensor_name = sub_features[i].name

        embedding.sprite.image_path = \
            'C:/Users/user/PycharmProjects/VariationalAutoEncoder/Tensorflow-DeconvNet-Segmentation/master.jpg'
        embedding.sprite.single_image_dim.extend([112, 112])

    # Saves a config file that TensorBoard will read during startup.
    projector.visualize_embeddings(tf.summary.FileWriter(LOG_DIR), config)