jfkuang / transformer_user_action

Transformer-based Realtime User Action Model for Recommendation at Pinterest

TransAct

Transformer-based Realtime User Action Model for Recommendation at Pinterest

TransAct is a PyTorch module for modeling user sequences using the Transformer architecture. It is designed to handle user behavior data with sequential interactions, such as recommendation systems or user activity prediction.

Paper

Xue Xia, Chantat Eksombatchai, Nikil Pancha, Dhruvil Deven Badani, Po- Wei Wang, Neng Gu, Saurabh Vishwas Joshi, Nazanin Farahpour, Zhiyuan Zhang, Andrew Zhai. 2023. TransAct: Transformer-based Realtime User Action Model for Recommendation at Pinterest.

Prerequisites

• Python 3.9.7 or higher: You can download and install Python from the official Python website at https://www.python.org/downloads/ or by using a package manager like conda or pip.

• torch 1.12: You can install torch 1.12 using pip with the following command:

  pip install torch==1.12
  

  Alternatively, if you are using conda, you can install torch 1.12 with the following command:

  conda install -c pytorch torch==1.12
  
  

Usage

python transact_code/test_run_transact.py