Create integration tests for session-based recommendation

Open gabrielspmoreira opened this issue 3 years ago • 2 comments

Description

Based on the training/eval script created in #806 , create unit/integration tests that ensures the stability of the API when setting the advanced hparams and also allowing to track in our CI accuracy or performance regressions

Additional context We have implemented similar unit/integration tests for retrieval models

The integration tests will cover four different architectures :

GPT-2 (CLM) - Item Id feature
BERT (MLM) - Item Id feature
XLNet (MLM) - Item Id feature
XLNET (MLM) - CONCAT - All features

Oct 13 '22 01:10 gabrielspmoreira

@sararb , please provide context on the 4 architecutres in the description

Apr 11 '23 17:04 viswa-nvidia

@gabrielspmoreira , please move this to another RMP ticket

Apr 25 '23 16:04 viswa-nvidia