From f099ce2fd02a1d064c443cd9b2a7dee4690afb8c Mon Sep 17 00:00:00 2001 From: Adrien <adrien.payen@student.uclouvain.be> Date: Fri, 24 May 2024 22:29:44 +0200 Subject: [PATCH] update readme --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 09d2ed8b..1af67a8d 100644 --- a/README.md +++ b/README.md @@ -97,14 +97,15 @@ Converts the predictions into a DataFrame and saves them as a CSV file. It then calls this function with specific parameters and prints the generated predictions. 2. #### ***content_based.ipynb*** -1. ***Feature Extraction Methods*** + +2.1. ***Feature Extraction Methods*** The system supports the following feature extraction methods: - `genre`: Extracts genres of the movies using TF-IDF vectorization. - `movie_year`: Extracts the release year of the movies. - `avg_rating`: Computes the average rating for each movie. - `title_length`: Computes the length of the movie title. -2. ***Regression Models*** +2.2. ***Regression Models*** The system supports the following regression models for predicting user ratings: - `linear_regression` -- GitLab