Skip to content

Lab-of-Infinity/Session-based-Recommendation-Using-SR-GNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

Session-based Recommendation Using SR-GNN

  • In this project we are building Session Based Recommendation system Using Graph Neural Network.

  • We have used real world E-comm data which available on Kaggle : Retail Rocket Dataset.

  • For Implementing Graph Neural Network we have utiliesd PyGeometric Package which build on top of Pytorch.

  • Author : Lokesh Baviskar

  • Email Id : [email protected]


  • On gross level Project flow as below:

    1. Constructing Session Graphs

    2. Learning Item Embeddings on Session Graphs

    3. Generating Session Embeddings

    4. Making Recommendation and Model Training

Data Set Information

  • We Have Utilised Retail Rocket Dataset ( Which is real world data collected from ecommerce platform ) for building Session Based Recommender.
  • You can download the data from 🔗this kaggle competition.
  • The behaviour data, i.e. events like clicks, add to carts, transactions, represent interactions that were collected over a period of 4.5 months.
  • A visitor can make three types of events, namely “view”, “addtocart” or “transaction”.
  • In total there are 2,756,101 events including 2,664,312 “view”, 69,332 “addtocart” and 22,457 “transaction” produced by 1,407,580 unique visitors.
  • The file with item properties includes 20,275,902 rows, i.e. different properties, describing 417,053 unique items.
  • We will only use the events.csv file.

image