Fix Label column meaning in Recommendation sample #1
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
I think that the meaning of
Label
column is incorrect in your postAccording to documentation for MatrixFactorizationTrainer Class
so the trainer expect input (sparse matrix) in the format
instead of this you provide matrix
A.[i,j]
should represent likelihood ofco-purchase
,1
meansco-purchase
and0
is most likely no.In this case trainer with factorize matrix and fill missing probabilities/likelyhood.
Page 28 of referenced paper also explain that https://www.csie.ntu.edu.tw/%7Ecjlin/talks/facebook.pdf
Output
With this fix, input matrix contain 0s and 1s and it is way more easier to factorize.
This is the result from your post
This is the result from this PR
RootMeanSquaredError
andMeanSquaredError
several order of magnitude smaller, but prediction for product 61 is similar =)