YongYu123 / PyTorch-Geometric-YooChoose

This is a tutorial for PyTorch Geometric on the YooChoose dataset

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

This is a tutorial for PyTorch Geometric on the YooChoose dataset

License:Apache License 2.0


Languages

Language:Jupyter Notebook 100.0%