神经关系推理(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.