search

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

Create integration tests for session-based recommendation #807

Open gabrielspmoreira opened 1 year ago

gabrielspmoreira commented 1 year ago

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 :

viswa-nvidia commented 1 year ago

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

viswa-nvidia commented 1 year ago

@gabrielspmoreira , please move this to another RMP ticket