Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,7 @@ Type: type
## Contributors

- Dileep Kishore
- Blessing Ibe


## Credits
Expand Down
4 changes: 4 additions & 0 deletions movie_recommender/movie_recommender.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
"""

from typing import List
from .utils import save, load, search
import warnings
warnings.filterwarnings("ignore", message="numpy.dtype size changed")
warnings.filterwarnings("ignore", message="numpy.ufunc size changed")
Expand Down Expand Up @@ -70,5 +71,8 @@ def movie_recommender(
recommender = KeywordSimilarity(metadata, credits, keywords)
elif method == "UserSimilarity":
recommender = UserSimilarity(metadata, ratings, links)
else:
raise ValueError()
print("Returning recommendations")
save(recommender)
return recommender(movie, nhits=nhits)