Add files via upload

Time-based-Activity-Recommender-System · Apr 4, 2017 · ff6c235 · ff6c235
1 parent 1499d7d
commit ff6c235
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/movieRatings_UI_run.py b/movieRatings_UI_run.py
@@ -197,7 +197,14 @@ def recommenderSystem(self):
         'Film-Noir', 'Horror', 'Musical', 'Mystery', 'Romance', 'Sci-Fi', 'Thriller', 'War', 'Western']
         items = pd.read_csv('ml-100k/u.item', sep='|', names=i_cols,encoding='latin-1')
         ind = np.argpartition(predictions_for_newuser, -1)[-5:]
-        for i in range(len(ind)):
+
+        ind2 = self.ratings_data['2movie_id'][0]
+        #print items['movie title'][ind2]
+        d = { 'movie_title': [ items['movie title'][ind2] ] }
+        df = pd.DataFrame(d)
+        df.to_csv('movie_reco.data' ,sep='\t',index=False, header=False)      
+
+        for i in range(1,len(ind)):
             ind2 = self.ratings_data['2movie_id'][i]
             #print items['movie title'][ind2]
             d = { 'movie_title': [ items['movie title'][ind2] ] }