NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations
https://nvidia-merlin.github.io/models/main/index.html
Apache License 2.0
262 stars 50 forks source link

Reproduce selected results from Transformers4Rec paper with Merlin Models API #806

Open gabrielspmoreira opened 2 years ago

gabrielspmoreira commented 2 years ago

Description

For Transformers4Rec, we have created a training/eval script for reproducing the paper experiments, that takes a set of hparams as command line arguments and a preprocessed dataset.

This task is about creating the training/eval script based on the original Transformers4Rec script using the Merlin Models API, which will:

Selected best results to reproduce with REES46 dataset (without features):

We should compare accuracy and runtimes for best trials, reported in this spreadsheet, in the paper and in the paper online appendix

viswa-nvidia commented 1 year ago

@gabrielspmoreira , please move this to another RMP ticket