NVIDIA-Merlin / NVIDIA-Merlin/Transformers4Rec
[FEA] Add an arg in the trainer class to check if it the task is NextItemPred or not
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 1.3k
- Forks
- 165
- Avg merge
- 1m
- Merged PRs (30d)
- 2
Description
🚀 Feature request
Add another argument in the trainer class to check if it the task is NextItemPred, otherwise it does have to return an error msg saying that trainer class is only for the Next item prediction task.
Motivation
Currently, some users are interested in using BinaryClassificationTask with TF4Rec but it is not clear for them if they should use Trainer Class or not. The Trainer class and its utilities were implemented for session-based and sequential tasks (i.e NextItemPredictionTask), so the class does not support tasks with explicit targets like BinaryClassification. This FEA is to avoid confusion and errors for users and help them to use properly TF4Rec for their BC tasks.
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.
Research direction
No file or test is named in the issue. Start by locating the Trainer class and the NextItemPredictionTask implementation or references, then inspect existing task validation and tests. Done means the trainer accepts the requested task check, reports an error for BinaryClassificationTask, and preserves supported next-item prediction behavior.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 38/100