File size: 2,834 Bytes
69ef591 028e00f 69ef591 028e00f 69ef591 028e00f 69ef591 028e00f 69ef591 028e00f ec885b3 028e00f 69ef591 724f895 69ef591 028e00f 69ef591 64827fd 69ef591 36444c9 724f895 9868478 69ef591 028e00f 9868478 69ef591 724f895 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
import gradio as gr
import numpy as np
import pandas as pd
from scipy.sparse import csr_matrix
from sklearn.neighbors import NearestNeighbors
def create_matrix(df):
N = len(df['userId'].unique())
M = len(df['movieId'].unique())
# Map Ids to indices
user_mapper = dict(zip(np.unique(df["userId"]), list(range(N))))
movie_mapper = dict(zip(np.unique(df["movieId"]), list(range(M))))
# Map indices to IDs
user_inv_mapper = dict(zip(list(range(N)), np.unique(df["userId"])))
movie_inv_mapper = dict(zip(list(range(M)), np.unique(df["movieId"])))
user_index = [user_mapper[i] for i in df['userId']]
movie_index = [movie_mapper[i] for i in df['movieId']]
X = csr_matrix((df["rating"], (movie_index, user_index)), shape=(M, N))
return X, user_mapper, movie_mapper, user_inv_mapper, movie_inv_mapper
def find_similar_movies(movie_id, X, k, metric='cosine', show_distance=False):
neighbour_ids = []
movie_ind = movie_mapper[movie_id]
movie_vec = X[movie_ind]
k += 1
kNN = NearestNeighbors(n_neighbors=k, algorithm="brute", metric=metric)
kNN.fit(X)
movie_vec = movie_vec.reshape(1, -1)
neighbour = kNN.kneighbors(movie_vec, return_distance=show_distance)
for i in range(0, k):
n = neighbour.item(i)
neighbour_ids.append(movie_inv_mapper[n])
neighbour_ids.pop(0)
return neighbour_ids
def recommend_movies(movie_name):
movie_id = [k for k, v in movie_titles.items() if movie_name.lower() in v.lower()]
if len(movie_id) == 0:
return "Movie not found. Please check the spelling and try again"
movie_id = movie_id[0]
similar_ids = find_similar_movies(movie_id, X, k=10)
recommendations = "\n".join([movie_titles[i] for i in similar_ids])
return recommendations
# Load data
ratings = pd.read_csv("ratings.csv")
movies = pd.read_csv("movies.csv")
n_ratings = len(ratings)
n_movies = len(ratings['movieId'].unique())
n_users = len(ratings['userId'].unique())
user_freq = ratings[['userId', 'movieId']].groupby('userId').count().reset_index()
user_freq.columns = ['userId', 'n_ratings']
mean_rating = ratings.groupby('movieId')[['rating']].mean()
lowest_rated = mean_rating['rating'].idxmin()
highest_rated = mean_rating['rating'].idxmax()
movie_stats = ratings.groupby('movieId')[['rating']].agg(['count', 'mean'])
movie_stats.columns = movie_stats.columns.droplevel()
X, user_mapper, movie_mapper, user_inv_mapper, movie_inv_mapper = create_matrix(ratings)
movie_titles = dict(zip(movies['movieId'], movies['title']))
# Set up Gradio interface
movie_name = gr.inputs.Textbox(default="Sweet November", label="Movie Name")
outputs = gr.outputs.Textbox(label="Recommended Movies", type="text")
iface = gr.Interface(
fn=recommend_movies,
inputs=movie_name,
outputs=outputs
)
iface.launch()
|