Skip to content

shrav-6/movies-tvshows-recommendation-system

Repository files navigation

Entip

Entertainment Recommendation System

Implementation

  1. The recommendation engine was implemented using two methods: content-based filtering and collaborative based filtering.
  2. For content-based filtering, count vectorization and cosine similarity were performed on the dataset using tags, these tags were filtered from the dataframe and the bag of words method was used.
  3. Collaborative based filtering was implemented using k-NN (k -Nearest Neighbor) machine learning model where the user’s rating is the key attribute.
  4. For multiple movie/tv shows recommendation, the single recommendation engine was used as an input to an algorithm which compares their similarity with the user’s input to obtain the appropriate suggestions.

    alt text
    Fig. Content based filtering

    alt text
    Fig. Multiple user input recommendation system

    alt text
    Fig. Collaborative based filtering

Test cases

Example test case for multi-user recommendation:
User’s Input:

  1. Urban Legend - Genres: Horror, Thriller
  2. Snatch – Genres: Thriller, Crime
  3. LOL - Genres: Drama, Comedy, Romance
  4. Devil – Genres: Horror, Mystery, Thriller
    alt text
    Fig. Output

Example test case for KNN:
User's Input: Iron Man
alt text
Fig. Output

Future Scope/ Enhancement:

  1. Completing the development of the web-application and integrating the recommendation engine with it, is of major concern.
  2. Broadening the fields of interest to documentaries as well as songs
  3. Optimizing the time complexity of the algorithm for faster recommendations.

About

Entertainment Recommendation System

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published