闭着眼睛复现Metapath2vec（Stellargraph）

from stellargraph import datasets
from stellargraph import IndexedArray
from stellargraph import StellarGraph
import pandas as pd
################################################################################
#命名可以没有顺序
author = IndexedArray(index=["a1","a2","a3","a4"])
paper = IndexedArray(index=["p1", "p2", "p3"])
venue = IndexedArray(index=["v1", "v2", "v3"])edges_AP = pd.DataFrame({"source": ["a1","a2","a3","a4"], "target": ["p1", "p2", "p3", "p3"]},index=[0,1,2,3]
)
edges_PV = pd.DataFrame({"source": ["p1", "p2", "p3"], "target": ["v1", "v2", "v3"]},index=[4,5,6]
)
#index的用处是为每一条边给一个唯一idGraph = StellarGraph(nodes={"author": author, "paper": paper,"venue": venue},edges={"AP":edges_AP, "PV":edges_PV})
print(Graph.info())################################################################################
#随机游走
walk_length = 100  # maximum length of a random walk to use throughout this notebook# specify the metapath schemas as a list of lists of node types.
metapaths = [["author", "paper", "author"],["author", "paper", "paper", "author"],["author", "paper", "venue","paper", "author"],
]from stellargraph.data import UniformRandomMetaPathWalk# Create the random walker
rw = UniformRandomMetaPathWalk(Graph)walks = rw.run(nodes=list(Graph.nodes()),  # root nodeslength=walk_length,  # maximum length of a random walkn=1,  # number of random walks per root nodemetapaths=metapaths,  # the metapaths
)print("Number of random walks: {}".format(len(walks)))################################################################################
#Word2Vec
from gensim.models import Word2Vec
model = Word2Vec(walks, vector_size=64, window=5, min_count=0, sg=1, workers=2, epochs=100)
model.save('word2vec.model')
#model = Word2Vec.load('word2vec.model')
################################################################################
vectors = model.wv.vectors #获取模型中全部的节点向量
words = model.wv.index_to_key #获取模型中全部的节点类型
vecs=model.wv['a1'] #取出一个节点的向量

附：安装stellargraph

pip install stellargraph

其他的包不用安装，stellargraph里面带了。

另外可能会出现 no module xxx（忘记叫啥了）。pip install xxx就行。

闭着眼睛复现Metapath2vec（Stellargraph）相关推荐

裸奔浏览器_躲进浏览器隐私模式就安全了吗？相当于闭着眼睛裸奔
「隐私模式」是浏览器的一个功能,记得小淙第一次看到这个充满安全感的名字时,心中一阵狂喜,如获珍宝,从此走上了不归路.多少个寂寞难耐的夜,我默默打开隐私模式,页面中间黑色的小人是我最默契的战友,我们相视 ...
先思再行闭着眼睛编程
摘要:解决问题最重要的习惯不是一直盯着屏幕和编写修改代码,某些时候,阻止你成功的东西恰恰会是过于努力.这时候你需要暂停一下,平缓你的思绪,换一种方法或许能带给你不一样的效果. 你会花多少时间思考如何编 ...
让用户闭着眼睛买东西可还行？雷军谈小米的目标...
5月29日,在2019中国电子商务大会上,小米科技董事长兼CEO雷军再次提及小米一直强调"感动人心,价格厚道的好产品",他表示在零售行业里面最重要的是我们怎么提供感动人心,价格厚道 ...
delphi framerect 画透明_闭上眼睛一起进入彩色画的梦境，幻彩花精灵，彩铅手绘教程...
点击"专栏"查看更多干货文章关注炫彩花精灵世上的花儿正在开放,美好的事情接踵而来,来和我一起闭上眼睛进入彩色花的梦境吧. 教程说明工具辉柏嘉蓝盒水溶60色自动铅笔 32K ...
MIT机器人闭上眼睛，靠触觉也爬得上凶险的楼梯 | 施工未完成
圆栗子发自凹非寺量子位出品 | 公众号 QbitAI 天黑请闭眼. 请各位玩家正确佩戴面具. 麻省理工的Cheetah (猎豹) 机器人,如今已经进化到了第三代. 比起前辈们,Chee ...
软件测试——闭着眼睛测试软件
闭着眼睛测试软件其实就是指动态黑盒子测试.它是动态的,因为程序正在运行--软件测试员充当客户使用它:它是黑盒子,因为测试时不知道程序如何工作--闭上眼睛. 对于软件测 ...
闭上眼睛时，会感触到那清风拂过脸颊的柔
在人生的旅途中,我在慢慢的找寻自己的那处风景--闭上眼睛时,会感触到那清风拂过脸颊的柔软.每个人的旅行都是一次美好的开始,回到原本我们曾经已经遗忘的那个角落,寻找那不在的人-- 曾经,我爬上山峰,去看 ...
你有一桶果冻，其中有黄色、绿色、红色三种，闭上眼睛抓取同种颜色的两个。抓取多少个就可以确定你肯定有两个同一颜色的果冻？（5秒-1分钟）
你有一桶果冻,其中有黄色.绿色.红色三种,闭上眼睛抓取同种颜色的两个. 抓取多少个就可以确定你肯定有两个同一颜色的果冻?(5秒-1分钟) 分析: 1.里面有三种颜色时,需要4个 2.里面有2种颜色时, ...
樱桃键盘驱动在哪下_闭着眼睛也能买的樱桃轴机械键盘——GANSS GS87C分享
昨天在社区里面溜达,看到B站发布 China Z-100百大产品榜单,推荐的时候居然看到GANSS GS87C的机械键盘,果然是我没买错,性价比高,价格是真的香~ 有兴趣的可以进去B站看一看,好多可以 ...

闭着眼睛复现Metapath2vec（Stellargraph）

闭着眼睛复现Metapath2vec（Stellargraph）相关推荐

最新文章

热门文章