tensorflow / tensorflow/recommenders

Ranking model predicting same value for all

Open
#377 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

question
Dominant language
Python
Stars
2k
Forks
300
PR merge metrics
No merged PRs in 30d

Description

Hi,

I am training a Ranking model with some basic numeric features and the target is either 0 or some value between 0-1.
So the model is training and just in 1 epoch it comes to a minimum loss and doesnot decrease more.
The main issue is observed when i predict with test data , i get all the predictions as 0. tried some changes in Layers but all leads to almost 99% values as 0
Where as same architecture when i trained using sequential model . its learning and getting atleast some values in prediction .
Not sure why with tfrs Ranking its some issue .

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.

Research direction

No files, tests, or entry points are named. Start by reproducing the Ranking model with the described numeric features and binary-or-continuous target, then compare its training and prediction behavior with the working Sequential model. Done means identifying why predictions collapse to zero and documenting or fixing the difference.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, tensorflow
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.