/PyTorch-Geometric-YooChoose

This is a tutorial for PyTorch Geometric on the YooChoose dataset

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Pytorch-Geometric-YooChoose

image

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.