DeepWalk & Node2Vec

DeepWalk & Node2Vec

维基百科词条DeepWalk图嵌入

import networkx as nx
import pandas as pd
import numpy as np
import random
from tqdm import tqdm
import matplotlib.pyplot as plt
%matplotlib inline

plt.rcParams["font.sans-serif"]=["SimHei"]
plt.rcParams["axes.unicode_minus"]=False

df = pd.read_csv("seealsology-data.tsv", sep="\t")

构建无向图

节点过多,省去可视化

生成随机游走序列

训练word2vec模型

分析word2vec结果

PCA降维可视化

可视化全部词条的二维embedding

可视化某个词条的二维embedding

可视化某些词条的二维embedding

TSNE降维可视化

可视化全部词条的二维embedding

导出TSNE降维到二维之后的embedding

可视化全部词条的三维enmbedding

导出TSNE降维到三维之后的embedding

<悲惨世界>人物关系-Node2Vec图嵌入

导入数据集

G = nx.les_miserables_graph