如何在 Torch Tensorboard Embedding 中添加多个元数据？

Question

我正在使用 torch==1.9.0 和 tensorboard==2.5.0。我想用 tensoraord 作为嵌入来跟踪数据，所以我是这样的：

data = np.random.poisson(lam=10.0, size=(4,4))
labels = ["A","A","B","B"]
ids = [1,2,3,4]
writer = SummaryWriter("/runs/")
writer.add_embedding(data,
                    metadata=labels)
writer.close()

但我只能将标签或 ID 添加为元数据，而不是组合的字典 {'ids':ids, 'lables':labels}。任何想法，如何解决这个问题？谢谢！

仅供参考：Tensorboard 文档只是将元数据描述为列表： https://pytorch.org/docs/stable/tensorboard.html

Answer 1

找到答案了。您可以通过添加元数据标题获得多个字段，并将元数据作为列表列表提供：

writer.add_embedding(data.values,
                    metadata=metadata,
                    metadata_header=["Name","Labels"])

参考：https://github.com/tensorflow/tensorboard/issues/61