首页 > 解决方案 > 如何指定张量板投影仪嵌入颜色

问题描述

我正在使用 tensorflow 的 (1.14) tensorboard 投影仪插件,将经过训练的数据嵌入到标记的类中。我希望能够指定张量板选择显示的颜色。

我查看了源代码,但无法找到指定颜色值的位置。

以下是我的代码的一些相关片段:

from tensorflow.contrib.tensorboard.plugins import projector
...
with tf.compat.v1.Session() as sess:
    saver = tf.compat.v1.train.Saver(tf_data)
    sess.run(initializers)
    saver.save(sess, '.')
    config = projector.ProjectorConfig()
    embedding = config.embeddings.add()
    embedding.metadata_path = 'labels.tsv'
    projector.visualize_embeddings(tf.compat.v1.summary.FileWriter('.'), config)

标签: pythontensorflowtensorboard

解决方案


也许看看https://github.com/tensorflow/tensorboard/issues/61

简而言之,您需要一个带有用于着色的列的 metadata.tsv 文件


推荐阅读