transformer_user_action
transformer_user_action copied to clipboard
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.12Alternatively, 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