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