NVIDIA-Merlin / Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.

https://nvidia-merlin.github.io/Transformers4Rec/main

Apache License 2.0

1.07k stars 142 forks source link

[FEA] Multi-task prediction support with Next-Item-Prediction #747

Closed vivpra89 closed 10 months ago

vivpra89 commented 10 months ago

🚀 Feature request

Current BinaryClassificationTask, RegressionTask can be used in multi-task scenario. Provide a way to include NextItem + Binary / Multiclass prediction

Motivation

There are cases where we want to predict NextItem for purchases and NextItem for Add To Carts. Those Cases arent supported currently

vivpra89 commented 10 months ago

https://github.com/NVIDIA-Merlin/Transformers4Rec/blob/30f0946447794d8edbfb32a3605c47a08213f98a/tests/unit/torch/test_trainer.py#L505

This shows how to use NextItemPred with classification task.