Recommendation System in Python
Recommendation System in Python
Algorithms -
Code –
import pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import cosine_similarity
# Load data
data = pd.read_csv('movies.csv')
title \
783 The Godfather
1512 The Godfather: Part II
1103 Casino
3509 Things to Do in Denver When
1246 Snatch
3094 Road to Perdition
2494 Scarface
1244 Following
2164 Dancer
2445 The Day of the Jackal
Short Explanation -
Next, the program defines a function that takes a movie title as input and returns a
list of the 10 most similar movies based on their plot summaries.
Finally, the program allows the user to enter a movie title and outputs a list of
recommended movies along with their similarity scores.
Code -
Algorithm -
import pandas as pd
from sklearn.metrics.pairwise import cosine_similarity
# Load data
ratings_data = pd.read_csv('ratings.csv')
# Example usage
user_id = 1
recommendations = get_recommendations(user_id, user_similarity,
user_item_matrix)
print("Top 10 recommended movies for user", user_id)
print(recommendations)
Output -