import torch import matplotlib.pyplot as plt # Vocabulary vocabulary = {'cat', 'dog', 'mouse'} # Embeddings in 2 dimensions embedding_dim = 2 # Create random embeddings for each word word_embeddings = { 'cat': torch.rand(embedding_dim), 'dog': torch.rand(embedding_dim), 'mouse': torch.rand(embedding_dim) } # Plot the embeddings fig, ax = plt.subplots() for word, embedding in word_embeddings.items(): ax.scatter(embedding[0], embedding[1]) ax.text(embedding[0], embedding[1], word) plt.xlabel('Dimension 1') plt.ylabel('Dimension 2') plt.title('Word Embeddings in 2D') plt.grid(True) plt.show()