from embedder.regression import Embedder
from embedder.assessment import visualize
import matplotlib.pyplot as plt

embedder = Embedder(embedding_dict)
embedder.fit(X_encoded, y)

embeddings = embedder.get_embeddings()
print(type(embeddings)) # a dictionary

visualize(embeddings['Store'])
plt.show()