Update app.py

app.py

@@ -212,6 +212,7 @@ cosine_sim = linear_kernel(tfidf_matrix, tfidf_matrix)
 
 
 smd = smd.reset_index()
+smd['title'] = smd['title'].apply(lambda x: str.lower(x))
 
 
 # In[24]:
@@ -224,25 +225,10 @@ indices = pd.Series(smd.index, index=smd['title'])
 # In[25]:
 
 
-tfidf_matrix.shape
 
 
-# In[34]:
 
 
-def get_recommendations(title):
-    if indices[title].shape ==():
-        idx = indices[title]
-    else:
-        idx = indices[title][0]
-    sim = cosine_sim    
-    sim_scores = list(enumerate(sim[idx]))
-    sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True)
-    sim_scores = sim_scores[1:5]
-    title_idx= [l[0] for l in sim_scores]
-    title_rec = [titles[i] for i in title_idx]
-    return title_rec
-
 
 # In[28]:
 
@@ -253,7 +239,6 @@ def get_recommendations(title):
 # In[27]:
 
 
-#get_recommendations('The Dark Knight',cosine_sim)
 
 
 # <b> <font size="3"> Adding the metadata to the rec system <font> <b>
@@ -388,6 +373,7 @@ cosine_sim2 = linear_kernel(count_matrix, count_matrix)
 
 
 smd = smd.reset_index()
+smd['title'] = smd['title'].apply(lambda x: str.lower(x))
 titles = smd['title']
 indices = pd.Series(smd.index, index=smd['title'])
 
@@ -401,20 +387,7 @@ indices = pd.Series(smd.index, index=smd['title'])
 # In[60]:
 
 
-def get_recommendations(title,sim):
-    if indices[title].shape ==():
-        idx = indices[title]
-    else:
-        idx = indices[title][0]
-    sim_scores = list(enumerate(sim[idx]))
-    sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True)
-    sim_scores = sim_scores[1:31]
-    title_idx= [l[0] for l in sim_scores]
-    title_rec = [titles[i] for i in title_idx]
-    return title_rec
-
 
-# In[62]:
 
 
 #get_recommendations('The Avengers',cosine_sim2)
@@ -426,6 +399,7 @@ def get_recommendations(title,sim):
 
 
 def improved_recommendations(title):
+    title = str.lower(title)
     idx = indices[title]
     sim_scores = list(enumerate(cosine_sim2[idx]))
     sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True)
@@ -442,7 +416,7 @@ def improved_recommendations(title):
     qualified['vote_average'] = qualified['vote_average'].astype('int')
     qualified['wr'] = qualified.apply(weighted_rating, axis=1)
     qualified = qualified.sort_values('wr', ascending=False).head(10)
-    return list(qualified['title'])
+    return list(qualified['title'].apply(lambda x: x.title()))
 
 
 # In[76]: