Update src/model/crb_crs/recommender/movie_recommender.py
Browse files
src/model/crb_crs/recommender/movie_recommender.py
CHANGED
@@ -534,10 +534,20 @@ class MovieRecommender(Recommender):
|
|
534 |
Args:
|
535 |
movie_id: Movie ID.
|
536 |
|
|
|
|
|
|
|
537 |
Returns:
|
538 |
Movie title.
|
539 |
"""
|
540 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
541 |
|
542 |
def replace_item_ids_with_recommendations(
|
543 |
self,
|
|
|
534 |
Args:
|
535 |
movie_id: Movie ID.
|
536 |
|
537 |
+
Raises:
|
538 |
+
KeyError: if movie title cannot be found.
|
539 |
+
|
540 |
Returns:
|
541 |
Movie title.
|
542 |
"""
|
543 |
+
try:
|
544 |
+
title = self.movie_mentions_df.loc[[int(movie_id)]]["title"].iloc[
|
545 |
+
0
|
546 |
+
]
|
547 |
+
except KeyError:
|
548 |
+
title = ""
|
549 |
+
logging.error(f"Movie title not found for movie ID {movie_id}.")
|
550 |
+
return title
|
551 |
|
552 |
def replace_item_ids_with_recommendations(
|
553 |
self,
|