hexiangnan / neural_collaborative_filtering

Neural Collaborative Filtering
Apache License 2.0
1.8k stars 655 forks source link

I can't get a good result in predicting ratings #46

Open Shiien opened 5 years ago

Shiien commented 5 years ago

i use mseloss instead of CrossEntropyLoss to predict ratings, the inputs are userId and itemId, and the target is an integer between 0 to 10. I transform it to 0,0.1,0.2...,1.0 and use sigmod as the last layer. but the output is very similar.why? please help me.