diff --git a/.gitignore b/.gitignore index 1c56d37..a64612b 100644 --- a/.gitignore +++ b/.gitignore @@ -7,3 +7,6 @@ raw_data/ # Cache __pycache__ .mypy_cache + +# ignore pickled +pickled* diff --git a/README.md b/README.md index f2f126e..323cef2 100644 --- a/README.md +++ b/README.md @@ -99,6 +99,7 @@ Type: type ## Contributors - Dileep Kishore +- Howard Fan ## Credits diff --git a/movie_recommender/movie_recommender.py b/movie_recommender/movie_recommender.py index 8e3bcaf..91752a9 100644 --- a/movie_recommender/movie_recommender.py +++ b/movie_recommender/movie_recommender.py @@ -16,6 +16,7 @@ from .plot_similarity import PlotSimilarity from .keyword_similarity import KeywordSimilarity from .user_similarity import UserSimilarity +from .utils import save, load, search def movie_recommender( @@ -70,5 +71,9 @@ def movie_recommender( recommender = KeywordSimilarity(metadata, credits, keywords) elif method == "UserSimilarity": recommender = UserSimilarity(metadata, ratings, links) + else: + raise ValueError('Method {} not supported'.format(method)) + print("Saving recommendations") + save(recommender) print("Returning recommendations") return recommender(movie, nhits=nhits)