PyTorch Geometric YooChoose

Pytorch-Geometric-YooChoose

Overview

This is the accompanying repository for my Medium article:

Hands-on Graph Neural Networks with PyTorch & PyTorch Geometric.

You will learn:

• how to implement custom Graph Convolutional layer with MessagePassing
• how to prepare data for training Graph Neural Networks
• how to build custom Graph Neural Networks

Files

• YooChooseClick.ipynb: All the code in the blog post lies here.
• YooChooseBuy.ipynb: The code I used to solve the second task of RecSys Challenge 2015.