Если вам нужен сюжет, похожий на этот: вам нужно уменьшить размерность ваших векторов слов до 2 измерений .
Итак, выдолжны применять к требуемым векторам слов алгоритм уменьшения размерности, такой как t-SNE (который также реализован в scikit-learn).
Для подобия оценки не достаточно; вам нужны целые векторы .
Здесь , есть хорошее руководство Kaggle по t-SNE для визуализации векторов слов. Вы можете настроить его, выбирая только те слова, которые вас интересуют.