【文件属性】：
文件名称：SELAR:PyTorch正式实施“具有异构图元路径的自我监督辅助学习”。 NeurIPS 2020年
文件大小：106KB
文件格式：ZIP
更新时间：2021-05-02 19:52:00
machine-learning graph-neural-networks pytorch-geometric Python
异构图元路径的自我监督辅助学习
该存储库是的实现。
黄达索（Dasol Hwang） * ，金永公园（Jinyoung Park） * ，孙永权，金敬敏，河正宇，金贤宇（Jyun-woo J.Kim），具有异构图元路径的自我监督辅助学习，神经信息处理系统的高级（NeurIPS 2020） 。
数据预处理
我们使用和数据集进行链接预测。 从此下载meta-paths标签（ meta_labels/ ）。
data/music/
ratings_final.npy ：由KGNN-LS发布预处理评价文件;
kg_final.npy ：知识图文件； meta_labels/
pos_meta{}_{}.pickle ：辅助任务的元路径肯定标签
neg_meta{}_{}.pickle ：辅助任务的元路径否定标签
data/book/
ratings_final.npy ：由Ripple
【文件预览】：
SELAR-main
----main_book.py(6KB)
----main_music.py(6KB)
----env.yml(3KB)
----sampler.py(30KB)
----utils.py(3KB)
----README.md(3KB)
----preprocess.py(3KB)
----evaluate.py(1KB)
----inits.py(928B)
----base_model.py(8KB)
----model.py(11KB)
----Figure_Main.png(97KB)
----train.py(17KB)