Загрузка данных


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']))