Skip to content

Machine learning powered model to recommend movies based on your interests written in python.

License

Notifications You must be signed in to change notification settings

kamaravichow/movie-recommendation-python

Repository files navigation

movie-recommendation-engine

Machine learning powered model to recommend movies based on your interests written in python.

Libraries :

  • sklearn.feature_extraction.text.CountVectorizer :

Convert a collection of text documents to a matrix of token counts

  • sklearn.metrics.pairwise.cosine_similarity(X, Y=None, dense_output=True) :

Compute cosine similarity between samples in X and Y. Cosine similarity, or the cosine kernel, computes similarity as the normalized dot product of X and Y:

Notes :

def combine_features(row):
	try:
	  return row['keywords']+ " "+row['cast']+" "+row['genres']+" "+row['director']
	except:
		print("Error", row)

This method is used because the kewords in the dataset has a "NaN" in it. We need to replace it with an empty string (You can do whatever you want). We make a for loop for that.

for feature in features:
	df[feature] = df[feature].fillna('') 

Simple system to break the loop after we get 50 suggestions

i=0
for movie in sorted_similar_movies:
	print(get_title_from_index(movie[0]))
	i=i+1
	if i>50:
		break

Usage : Unzip the dataset zip file and then use it.

About

Machine learning powered model to recommend movies based on your interests written in python.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages