/

Neural Network Recommendation System

Machine Learning | PyTorch

Project

Overview

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