Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Collaborative filtering example #255

Open
sgaseretto opened this issue Aug 6, 2022 · 2 comments
Open

Collaborative filtering example #255

sgaseretto opened this issue Aug 6, 2022 · 2 comments

Comments

@sgaseretto
Copy link

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.

@ToucheSir
Copy link
Member

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.

@sgaseretto
Copy link
Author

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants