如何在 TensorBoard 嵌入可视化中为不同的标签使用不同的颜色?



我正在使用张量板可视化句子嵌入。我为每个句子嵌入都有标签。如何为每个标签设置颜色? 例如

embedding vector                        Labels
[0.2342 0.2342 0.234 0.8453]             A
[0.5342 0.9342 0.234 0.1453]             B
[0.7342 0.0342 0.124 0.8453]             C
[0.8342 0.5342 0.834 0.5453]             A

我能够可视化嵌入向量,其中每一行都由其标签标记。我还想设置颜色,以便我看到具有相同标签的点将具有相同的颜色。就像所有的"A"都是红色的,"B"会是绿色的,"C"会是蓝色的等等?

我在谷歌上搜索,但没有得到任何样本。 任何人都可以分享一些代码来完成它吗?

提前谢谢你。

应该有一个可以使用的下拉列表颜色。 如果未显示,其中一个可能的原因可能是您有超过 50 个唯一标签,这是当前 tensorflow 代码中的硬编码限制。

有关详细信息,请参阅此线程。

https://github.com/tensorflow/tensorboard/issues/61

最新更新