• Introduction
• Usage
• • Overall
  • Model
  • Acknowledge

Introduction

This repository implements models RoBERTa4Item. Use RoBERTa4Item model to learn the dynamic embedding information of item.

Usage

Overall

Run main.py with arguments to train and/or test you model. There are predefined templates for all models.

On running main.py, it asks you whether to train on MovieLens-1m or MovieLens-20m. (Enter 1 or 20)

python main.py --template train_bert

Model

Pre-tained model placed in ‘model_save’.

Acknowledge

This work was supported by the National Key R&D Program of China under Grant No. 2020AAA0103804(Sponsor: Hefu Liu). This work belongs to the University of science and technology of China.