Generate a heat map from word embeddings using universal sentence encoder