NVIDIA-Merlin / NVIDIA-Merlin/Merlin

[RMP] Support of advanced sequential tasks and the definition of examples

Open
#472 0 comments 0 reactions 3 assignees View on GitHub

@edknv is already working on this.

Since Aug 16, 2022.

P1 roadmap
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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.