FastAI.jl icon indicating copy to clipboard operation
FastAI.jl copied to clipboard

Collaborative filtering example

Open sgaseretto opened this issue 1 year ago • 2 comments

How can collaborative filtering (with probabilistic matrix factorization) be implemented with FastAI.jl? Are there any snippets of this or plans to add it soon? I haven't seen examples of this even with Flux.jl so I'm a little bit lost, only MLP and Convnets. Also, I'm learning Julia and would love to contribute to this package, If there are some hints on how to do this, I'll gladly try to implement it and open a PR.

sgaseretto avatar Aug 06 '22 22:08 sgaseretto

I don't know of anyone working on collaborative filtering in Julia (the ML community is quite small), but perhaps others have. One way to get the ball rolling would to be port over a Python implementation. If you have a good grasp of that side, I'd be happy to help with the translation.

ToucheSir avatar Aug 06 '22 22:08 ToucheSir

I'll research a little bit more about Flux.jl, so I can get a better understanding on how it builds its computational graph, because it is basically a dot product between a user embedding and an item embedding. Here is an excellent example of how it can be implemented using pytorch.

sgaseretto avatar Aug 06 '22 22:08 sgaseretto