cosanlab / neighbors

A package to perform collaborative filtering on emotion datasets.
https://cosanlab.github.io/neighbors
MIT License
10 stars 8 forks source link

[WIP] Potential enhancement to multiplicative updating #33

Open ejolly opened 3 years ago

ejolly commented 3 years ago

This tries to use the approach in Blondel et al and Zhu, 16 for using a weight/mask matrix to ignore missing values while doing the Lee & Seung multiplicative update. The latter implementation can be found in a git repo here

It initially seemed to work well for some data where out-of-sample performance increased, but for other data it performed horrendously where predictions weren't even on the same scale as the user ratings. I got a little excited and merged it into master in #31, but given how finicky it seems to be, I've unmerged it now and I'm leaving it in this branch/PR for posterity.