Skip to content

In this repository, a recommendation system based on the paper "Behavior Sequence Transformer for E-commerce Recommendation in Alibaba" is implemented using Pytorch framework.

License

Notifications You must be signed in to change notification settings

F-Yousefi/RecSys-BST-Pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 

Repository files navigation

banner.PNG header.jpg MovieLens.org | Bob Liu

Movie Recommendation System

Behavior Sequence Transformer- Pytorch


Abstrsct

Recommendation systems play an essential role in our lives; however they are almost ignored academically. In this project, I will create a movie recommendation system model based on Movielens dataset. The neat idea behind this project is inspired by the paper Behavior Sequence Transformer for E-commerce Recommendation in Alibaba that can be found through this link. In mentioned article, the author explains why this model and architecture responds better than any other RecSys before it. This architecture uses the feature that almost all the other RecSys ignored, the behavior sequence of each user over a period of time. For example, in our case, when a user has never seen a horror movie, definitely "The Silence of Lambs" is not the movie that a good RecSys recommends to him/her. Additionally, a user might decide to see movies of other genres too. For example, after watching hundreds of drama movies, he/she decided to watch comedy movies, in this case all the previous generation of RecSys might fail to understand the pattern, but not this one.

Dataset

Movielens

GroupLens Research has collected and made available rating data sets from the MovieLens web site https://movielens.org. The data sets were collected over various periods of time, depending on the size of the set. We have chosen the dataset with one million records for this project. MovieLens 1M consists 1 million ratings from 6000 users on 4000 movies.

Download

The dataset will be automatically downloaded as you run the main .ipynb file.

Architecture

Papers

This project is implemented based on two breakthrough papers Behavior Sequence Transformer for E-commerce Recommendation in Alibaba and Attention is all you need. For more information and acquiring a better understanding of the project, you can consult these papers.

Graphical Architecture

As you can see, the diagram below is representing the whole architecture of the model used in this project.

Left Graph : Graphical Architecture

Right Table: All the layers of the model


graphical_architecture.jpg table_architecture.PNG

References

This project is built thanks to the valueable information that I have found through the references that i have listed below.

Qiwei Chen, Huan Zhao, Wei Li, Pipei Huang, and Wenwu Ou. 2019. Behavior sequence transformer for e-commerce recommendation in Alibaba. 1

Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N. Gomez, Lukasz Kaiser, Illia Polosukhin. Attention Is All You Need.2

About

In this repository, a recommendation system based on the paper "Behavior Sequence Transformer for E-commerce Recommendation in Alibaba" is implemented using Pytorch framework.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published