import networkx as nx
from node2vec import Node2Vec
import matplotlib.pyplot as plt
G = nx.Graph()
G.add_edges_from([
('A', 'B'), ('A', 'C'),
('B', 'D'), ('C', 'D'),
('D', 'E'), ('E', 'F'),
('F', 'G'), ('E', 'G'),
('G', 'H'), ('H', 'I')
])
# Получите эмбеддинги для всех узлов с помощью Node2Vec (dimensions=8)
node = Node2Vec(G, dimensions=8)
# Выведите вектор узла 'D'
model = node.fit(window=10, min_count=1, batch_words=4)
# Постройте визуализацию графа
print("Вектор узла 'D': [", list(model.wv['D']))