• Stars
    star
    304
  • Rank 137,274 (Top 3 %)
  • Language
    Jupyter Notebook
  • License
    Apache License 2.0
  • Created over 5 years ago
  • Updated over 1 year ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

This is a tutorial for PyTorch Geometric on the YooChoose dataset

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.