Scribd
Upload
17 views2 pages

Movie Recommend

Program

Uploaded by

divyahs2005
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
17 views2 pages

Movie Recommend

Program

Uploaded by

divyahs2005
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as PDF, TXT or read online on Scribd
You are on page 1/ 2

import streamlit as st

import pandas as pd
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.metrics.pairwise import cosine_similarity

# Load movie data


def load_data():
movies = {
'title': [
'Avatar', 'Titanic', 'The Avengers', 'The Dark Knight',
'Shutter Island', 'Inception', 'Interstellar', 'Iron Man',
'Thor', 'The Martian'
],
'genres': [
'Action Adventure Sci-Fi', 'Drama Romance', 'Action
Adventure Sci-Fi',
'Action Crime Drama', 'Mystery Thriller', 'Action Adventure
Sci-Fi',
'Adventure Drama Sci-Fi', 'Action Adventure Sci-Fi',
'Action Fantasy',
'Adventure Drama Sci-Fi'
]
}
return pd.DataFrame(movies)

# Compute similarity matrix


def compute_similarity(data):
count_vectorizer = CountVectorizer()
genre_matrix = count_vectorizer.fit_transform(data['genres'])
return cosine_similarity(genre_matrix, genre_matrix)

# Recommend movies
def recommend_movies(movie_title, data, similarity_matrix):
if movie_title not in data['title'].values:
return ["Movie not found. Please check your input."]
index = data[data['title'] == movie_title].index[0]
similarity_scores = list(enumerate(similarity_matrix[index]))
similarity_scores = sorted(similarity_scores, key=lambda x: x[1],
reverse=True)
top_movies = [data['title'][i[0]] for i in similarity_scores[1:6]]
return top_movies

# Streamlit UI
st.title("Movie Recommendation System")
st.write("Enter a movie title, and we'll recommend similar movies.")

data = load_data()
similarity_matrix = compute_similarity(data)

movie_title = st.text_input("Enter a movie title:")


if st.button("Recommend Movies"):
recommendations = recommend_movies(movie_title, data,
similarity_matrix)
st.write("### Recommended Movies:")
for rec in recommendations:
st.write("- ", rec)

You might also like