子神经网络 NeurIPS 2020论文存储库: 作者: , ,, 要使用SubGNN,请执行以下操作: 安装环境 准备数据 在config.py修改PROJECT_ROOT 修改适当的config.json文件 训练和评估SubGNN 安装环境 我们提供了一个yml文件,其中包含SubGNN的必要软件包。 一旦安装了 ,就可以创建如下环境: conda env create --file SubGNN.yml 准备数据 通过(1)下载我们提供的数据集或按照prepare_dataset文件夹README中的步骤来为SubGNN准备数据,(2)生成合成数据集或(3)格式化您自己的数据。 真实数据集:我们将发布四个新的真实数据集:HPO-NEURO,HPO-METAB,PPI-BP和EM-USER。 您可以 从Dropbox下载这些文件。 您应该解压缩文件夹并将config.py的P
2025-10-21 11:52:49 87KB embeddings graph-neural-networks Python
1
PyTextGCN 对TextGCN的重新实现。 此实现使用Cython进行文本到图形的转换,因此速度相当快。 图形和GCN基于库。 要求 该项目的构建具有: 的Python 3.8.5 Cython 0.29.21 CUDA 10.2(GPU支持可选) scikit学习0.23.2 pytorch 1.7.0 火炬几何1.6.3 海湾合作委员会9.3.0 nltk 3.5 scipy 1.5.2 至少Text2Graph模块也应该与这些库的其他版本一起使用。 安装 cython编译可以从项目的根目录执行: cd textgcn/lib/clib && python setup.py build_ext --inplace 用法 要从称为X的字符串列表(每个字符串包含一个文档的文本)中计算出图形,请创建名为y的标签列表以及测试索引test_idx的列表,只需运行:
1
研究@ Magic Leap(CVPR 2020,口腔) SuperGlue推理和评估演示脚本 介绍 SuperGlue是在Magic Leap完成的2020 CVPR研究项目。 SuperGlue网络是一个图形神经网络,结合了最佳匹配层,该层经过训练可以对两组稀疏图像特征进行匹配。 此存储库包含PyTorch代码和预训练权重,用于在关键点和描述符之上运行SuperGlue匹配网络。 给定一对图像,您可以使用此存储库在整个图像对中提取匹配特征。 SuperGlue充当“中端”,在单个端到端体系结构中执行上下文聚合,匹配和过滤。 有关更多详细信息,请参见: 全文:PDF: 。 作者: Pa
1
神经关系推理(NRI) 用于交互系统的图神经网络 给定节点的时间序列数据,NRI模型会将未来的节点状态和节点之间的基础抵销关系预测为边缘。 这是Chainer中神经关系推理(NRI)的再现作品。 作者的原始实现可在此处找到: 。 请参阅本文的详细信息: 交互系统的神经关系推断。 Thomas Kipf *,Ethan Fetaya *,Kuan-Chieh Wang,Max Welling,Richard Zemel。 :平等贡献) 数据集 粒子物理模拟数据集 cd data python generate_dataset.py 训练 粒子物理模拟数据集 python train.py --gpu 0 可视化结果 python utils/visualize_results.py \ --args-file results/2019-01-22_10-20-25_0/args.
2023-03-28 18:42:31 1.09MB deep-learning chainer graph-neural-networks Python
1
用于学习分子图的分层消息间传递 这是用于学习分子图的分层消息间传递的 PyTorch 实现,如我们的论文中所述: Matthias Fey、Jan-Gin Yuen、Frank Weichert:(GRL+ 2020) 要求 (>=1.4.0) (>=1.5.0) (>=1.1.0) 实验 可以通过以下方式运行实验: $ python train_zinc_subset.py $ python train_zinc_full.py $ python train_hiv.py $ python train_muv.py $ python train_tox21.py $ python train_ogbhiv.py $ python train_ogbpcba.py 引用 如果您在自己的工作中使用此代码,请引用: @inproceedings{Fey/etal/2020,
1
HiAGM:层次结构文本分类的层次结构感知全局模型 该存储库实现了层次结构感知的结构编码器,用于标签空间和文本特征之间的相互交互。 这项工作已被接受为ACL 2020中的长篇论文《的》。该存储库中提出了NYTimes(《纽约时报》)和WoS(Web of Science)的数据集拆分。 层次感知全局模型 具有层次结构意识的全局模型利用预先定义的层次结构的先验知识改进了常规文本分类模型。 项目文件夹由以下部分组成: config:配置文件(json格式) 数据:数据目录,可以在配置文件中更改(带有示例数据) data_modules:数据集/ DataLoader /整理器/ Vocab 助手:配置/ Hierarchy_Statistic /记录器/实用工具 模型:StructureModel / EmbeddingLayer / TextEncoder / TextPropaga
1
半监督分层递归图神经网络用于城市范围内的停车位可用性预测 这是SHARE体系结构的Pytorch实现,如论文《。 如果您在研究中利用SHARE模型,请引用以下内容: @article{zhang2019semi, title={Semi-Supervised Hierarchical Recurrent Graph Neural Network for City-Wide Parking Availability Prediction}, author={Zhang, Weijia and Liu, Hao and Liu, Yanchi and Zhou, Jingbo and Xiong, Hui}, booktitle={Proceedings of the Thirty-Fourth AAAI Conference on Artificial Intelligen
1
纸 论文“深度融合集群网络”的源代码 图W.涂,周S.,刘X.,郭X,蔡Z. 被AAAI2021接受。 安装 克隆此仓库。 git clone https://github.com/WxTu/DFCN.git Windows 10或Linux 18.04 的Python 3.7.5 脾气暴躁的1.18.0 斯克莱恩0.21.3 火炬视觉0.3.0 Matplotlib 3.2.1 准备 我们总共采用了六个数据集,包括三个图形数据集(ACM,DBLP和CITE)和三个非图形数据集(USPS,HHAR和REUT)。 要在这些数据集上训练模型,请从(访问代码:4622)或下载它们。 代码结构与用法 在这里,我们提供了PyTorch中的深度融合集群网络(DFCN)的实现,以及DBLP数据集上的执行示例(由于文件大小的限制)。 该存储库的组织方式如下: load_data.py
1
EGNN-Pytorch(WIP) 中的实现。 最终可用于Alphafold2复制。 安装 $ pip install egnn-pytorch 用法 import torch from egnn_pytorch import EGNN layer1 = EGNN ( dim = 512 ) layer2 = EGNN ( dim = 512 ) feats = torch . randn ( 1 , 16 , 512 ) coors = torch . randn ( 1 , 16 , 3 ) feats , coors = layer1 ( feats , coors ) feats , coors = layer2 ( feats , coors ) # (1, 16, 512), (1, 16, 3) 带边 import torch from egnn_pytorch impo
1
基于原始分子图的神经网络属性预测 该代码是在阿斯利康进行的两项工作的基础: 我的硕士学位论文 Me和Michael Withnall的论文《 的中,本文提到的三个模型引用了以下代码和论文的模型: SELU-MPNN-> GGNN AMPNN-> AttentionGGNN GGNN EMNN-> EMN 论文的技术细节更为丰富,但并未经过同行评审,其中包含错误生成的ESOL数据集结果。 本文包含了更详尽,更仔细地生成的结果集。 相关工作 最重要的四篇相关论文是: 提供了一个图神经网络作为本工作以及以下论文的基线 定义了图表的神经网络的MPNN框架,在该代码实现为抽象类SummationMPNN 提供了一种用于节点分类的模型,该模型具有消息聚合步骤,该步骤不适合MPNN框架,但可以适合作为抽象AggregationMPNN类实现的更通用的框架,在计算上可以看作是较轻的变体。当前
2022-05-02 18:36:03 38KB Python
1