【文件属性】：
文件名称：MotifGNN：论文的官方代码实现基于Motif的图神经网络对在线约会的相互推荐
文件大小：23KB
文件格式：ZIP
更新时间：2021-02-19 15:01:25
Python
MotifGNN
作者
纸的官方代码实现
环境环境
Ubuntu 20.04
python == 3.6
显卡：RTX 2080
CUDA：10.1
数据库：Neo4j
Python功能
pytorch==1.6.0
pandas
neo4j
sklearn
numpy
gensim
tqdm
下载代码
由于隐私问题，我们仅公开ssjy数据集。 您可以在下载数据集并解压缩到“数据”文件夹。
数据预处理
将图加载到neo4j
sudo neo4j-admin import --database=sjjy.db --nodes:User=data/sjjy_data/users.csv --relationships=data/sjjy_data/relations.csv --id-type=INTEGER
生成训练和测试数据
python3 data/prepare_dat
【文件预览】：
MotifGNN-master
----data()
--------.gitkeep(0B)
----our_model()
--------eval.py(4KB)
--------mymodel_config.py(894B)
--------recommend_model.py(5KB)
----README.md(1KB)
----src_sjjy()
--------eval_mymodel.py(3KB)
--------motif_random_walk.py(10KB)
--------generate_walk_path.py(8KB)
--------utils.py(2KB)
--------pipline_config.py(3KB)
--------train_pipline.py(650B)
--------train_emb.py(2KB)
--------data_preprocess.py(4KB)
--------model.py(4KB)
--------train_config.py(1001B)
--------train.py(9KB)
----checkpoint()
--------.gitkeep(0B)