Visualize embeddings in tensorboard: using Tensorboard projector

use_tb_projector.py

import numpy as np
import tensorboard as tb
from torch.utils.tensorboard import SummaryWriter
import tensorflow as tf
tf.io.gfile = tb.compat.tensorflow_stub.io.gfile

writer = SummaryWriter()
writer.add_embedding(<embeddings>, <labels>)  # (doc_vector, Y_true[:TRAIN_SIZE_EVALS[-3]])
writer.close()
%load_ext tensorboard
%tensorboard --logdir=runs

# Taken from this answer: https://stackoverflow.com/a/41177133