Геометрическая визуализация косинусного подобия ⇐ Python

[Anonymous]

Я рассчитал косинусное сходство между двумя документами очень простым способом, используя векторизацию TF-IDF в Python.
Но я хочу визуализировать документы в виде векторизованного графика в трехмерном пространстве. Вот так:


Вот код, который я использовал для расчета косинусного сходства:
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import cosine_similarity
import matplotlib.pyplot as plt
import numpy as np

# Sample documents
documents = [
"This is a sample document.",
"This document is another example."
]

# Vectorizing the documents
vectorizer = TfidfVectorizer()
tfidf_matrix = vectorizer.fit_transform(documents)

# Calculating cosine similarity
cosine_sim = cosine_similarity(tfidf_matrix[0:1], tfidf_matrix[1:2])
print("Cosine Similarity:", cosine_sim)


Подробнее здесь: https://stackoverflow.com/questions/791 ... similarity