PyTorch Geometric YooChoose

Pytorch-几何-YooChoose

概述

这是我的 Medium 文章的随附存储库：

使用 PyTorch 和 PyTorch Geometric 实践图形神经网络。

您将学到：

• 如何使用MessagePassing实现自定义图卷积层
• 如何准备训练图神经网络的数据
• 如何构建自定义图神经网络

文件

• YooChooseClick.ipynb：博客文章中的所有代码都位于此处。
• YooChooseBuy.ipynb：我用来解决 RecSys Challenge 2015 第二个任务的代码。