diff --git a/README.md b/README.md index 09d2ed8b3f0f56ee0dc577a8ad48b77603088742..1af67a8d07e98807c88bf7726273eda6ab5c42dc 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`