This is a POC of a movie recommendation system. It uses a Content-Based approach by mapping the movie features into a 2d space using UMAP, then with a SVR it predicts the 5 star rating based on the user's previus ratings.
- Copy the env file and change the values accordingly.
cp .env.example .env- Then start the services with:
make up-
Open http://localhost:5173 in your browser.
-
Create an account
-
Rate the movies. Firstly, the system will recommend randomly. From 3 rated movies it will recommend based on your preference.
You can start the tests using
make testYou can evaluate the model by running:
make trainIt uses the function cross_val_score to evaluate the model accuracy.