Project
Implemented a simple movie recommendation system by treating the problem as a regression problem and implemented a neural network that trains embeddings of the users. Used a dataset from MovieLens: https://grouplens.org/datasets/movielens/. Learned about performance metrics for recommendation systems such as mean average precision to evaluate my trained model.
Technologies
Python
PyTorch
Back