diff --git a/.gitignore b/.gitignore index 1c56d37..3923258 100644 --- a/.gitignore +++ b/.gitignore @@ -7,3 +7,6 @@ raw_data/ # Cache __pycache__ .mypy_cache + +# ignore recommender object +recommender_saved diff --git a/README.md b/README.md index f2f126e..5c24e6f 100644 --- a/README.md +++ b/README.md @@ -99,6 +99,8 @@ Type: type ## Contributors - Dileep Kishore +- Callen Bragdon +- Conor Shea(It's all coming together) ## Credits diff --git a/movie_recommender/movie_recommender.py b/movie_recommender/movie_recommender.py index 8e3bcaf..339a805 100644 --- a/movie_recommender/movie_recommender.py +++ b/movie_recommender/movie_recommender.py @@ -16,7 +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( metadata_file: str, @@ -70,5 +70,8 @@ def movie_recommender( recommender = KeywordSimilarity(metadata, credits, keywords) elif method == "UserSimilarity": recommender = UserSimilarity(metadata, ratings, links) + else: + raise("Value Error; Unsupported method. use --help to learn more") print("Returning recommendations") + save(recommender(movie,nhits=nhits),"recommender_saved") return recommender(movie, nhits=nhits)