NVIDIA-Merlin / NVIDIA-Merlin/Merlin
[RMP] Support of advanced sequential tasks and the definition of examples
@edknv is already working on this.
Since Aug 16, 2022.
- Dominant language
- Python
- Stars
- 907
- Forks
- 129
- PR merge metrics
- No merged PRs in 30d
Description
Problem:
Following the integration of session-based support (#451, #452), the missing component would be providing recsys practitioners with all the necessary blocks for advanced use-cases such as multi-task learning for item-level predictions and negative sampling support for the NextItemPrediction task.
Goal:
- Implement the blocks and negative sampling strategies required for training with item-level prediction tasks.
Constraints:
- #452 should be finalized before starting this work
Starting Point:
-
Refactory of the prediction blocks to support the ContrastiveLearning class.
-
Extend regression and classification prediction block to support sequential tasks (item-level)
-
Implement negative sampling support for session-based tasks
Note: support bpr, bpr-max, top1, and top1-max as defined in the GRU4Rec paper. -
Definition of advanced example notebooks showcasing multi-task learning for the session-based recommendation.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.