Merlin icon indicating copy to clipboard operation
Merlin copied to clipboard

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

Open sararb opened this issue 3 years ago • 0 comments

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.

sararb avatar Jul 18 '22 12:07 sararb