Social-STGCNN:用于人类轨迹预测的社会时空图卷积神经网络 阿卜杜拉·穆罕默德·昆茜 Mohamed Elhoseiny **,Christian Claudel ** **平等建议 阅读全文,在 社会-STGCNN 我们提出了社会时空图卷积神经网络(Social-STGCNN),该模型将人类轨迹预测问题建模为时空图。 我们的结果表明,与先前报道的方法相比,最终位移误差(FDE)较现有技术提高了20%,平均位移误差(ADE)的改进比参数减少了8.5倍,推理速度提高了48倍。 此外,我们的模型具有较高的数据效率,仅使用20%的训练数据就超出了ADE指标上的现有技术水平。 我们提出了一个核函数,将行人之间的社交互动嵌入邻接矩阵中。 仅使用7.6K参数,我们的模型推断速度为0.002s /帧(500Hz)。 Citaion 您可以使用以下方法引用我们的论文: @inprocee
1
KGCN-火炬 这是 ( )的Pytorch实现: 推荐系统的知识图卷积网络王宏伟,赵M,谢星,李文杰,郭敏仪。 在2019年网络会议论文集(WWW 2019)中 数据集 电影 电影的原始分级文件太大,无法包含在此仓库中。 首先对评级数据进行分类 $ wget http://files.grouplens.org/datasets/movielens/ml-20m.zip $ unzip ml-20m.zip $ mv ml-20m/ratings.csv data/movie/ 音乐 没事做 其他数据集 如果要使用自己的数据集,则需要准备2个数据。 评分数据 每行应包含(user-item-rating) 在此回购中,它是pandas数据框结构。 (看看data_loader.py ) 知识图 每个三元组(头-关系尾)由知识图组成 在此仓库中,它是字典类型。 (看看data_
1
text_gcn 本文中Text GCN的实现: 梁耀,毛成胜,罗源。 “图卷积网络用于文本分类。” 在第33届AAAI人工智能会议(AAAI-19)中 要求 Python 2.7或3.6 Tensorflow> = 1.4.0 再现结果 运行python remove_words.py 20ng 运行python build_graph.py 20ng 运行python train.py 20ng 在为其他数据集生成结果时,将上述3个命令行中的20ng更改为R8 , R52 , ohsumed和mr 。 输入数据示例 /data/20ng.txt表示文档名称,培训/测试组,文档标签。 每行都是一个文档。 /data/corpus/20ng.txt包含每个文档的原始文本,每行对应/data/20ng.txt的相应行 prepare_data.py是准备自己的数据的示例,请注意,文档或句子中的“ \ n”已删除。 归纳版 文本GCN的归纳版本是 ,其中培训过程中未包括测试文档。
1
双水平图神经网络 总览 此存储库包含用于药物相互作用预测的双层图神经网络的代码。 有关详细信息,请参阅我们的论文 用于药物相互作用预测的双层图神经网络。 白云生*,顾坚*,孙宜州,王伟。 ICML 2020图形表示学习及超越(GRL +)研讨会 。 模型 我们引入Bi-GNN建模生物链接预测任务,例如药物-药物相互作用(DDI)和蛋白质-蛋白质相互作用(PPI)。 以药物-药物相互作用为例,使用机器学习的现有方法要么仅利用药物之间的链接结构而不使用每个药物分子的图形表示,要么仅利用单个药物化合物的结构而不对高级DDI使用图形结构图形。 我们方法的关键思想是从根本上将数据视为双层图,其中最高
1