Changing data from 2k titles to 26k titles. Workaround for ease and cbf

app.py

@@ -5,13 +5,20 @@ from easeBased_model import *
 from contentBased_model import *
 
 
+
 if "user_preferences" not in st.session_state:
     st.session_state["user_preferences"] = {}
 
 # if "user_titles" not in st.session_state:
 #     st.session_state["user_titles"] = []
 
-data = pd.read_pickle("data/2k_titles.pkl")
+# if "26k_titles" not in st.session_state:
+#     st.session_state['26k_titles'] = data
+
+data = pd.read_pickle("data/list_of_all_titles.pkl")
+ease_ids = list(
+    pd.read_pickle("data/dict_of_ease_ids.pkl").keys())
+
 
 def games_recomm(preferences_id):
     if "rs" in st.session_state:
@@ -30,12 +37,25 @@ def games_recomm(preferences_id):
             'app_id': preferences_id,
             'is_recommended': pref_value
         })
+        ease_df = pred_df[pred_df['app_id'].isin(ease_ids)]
 
-        res_ease = ease_model(pred_df=pred_df, k=100)
         res_cbf = cbf_model(pred_df=pred_df, k=100)
+        res_ease = ease_model(pred_df=ease_df, k=100)
+
         res = combine_hybrid_result(res_ease, res_cbf)
+        # res = cbf_model(pred_df=pred_df, k=100)
         
     # st.write(res)
+    st.write("Result of EASE:")
+    st.write(res_ease)
+    st.markdown("<br>", unsafe_allow_html=True)
+
+    st.write("Result of CBF:")
+    st.write(res_cbf)
+    st.markdown("<br>", unsafe_allow_html=True)
+
+    st.write("Result of Hybrid:")
+    st.write(res)
 
     if type(res) == ValueError:
         st.error("Recommendation failed. Please select with at least 3 games title.")
@@ -43,6 +63,8 @@ def games_recomm(preferences_id):
         return
     else:
         st.session_state['rs'] = res
+    # st.session_state['rs'] = res
+    st.session_state
 
     if len(st.session_state['rs']) >= 1:
         st.success(f"Go to result page to view top {len(st.session_state['rs'])} recommendations.")

contentBased_model.py

@@ -118,7 +118,8 @@ class KnnCBF:
         return pred_result
     
 def cbf_model(pred_df, k=10):
-    items = pd.read_csv("data/games_attributes.csv")
+    # items = pd.read_csv("data/games_attributes.csv")
+    items = pd.read_csv("data/all_games_attributes.csv")
 
     cbf = KnnCBF(items)
     res = cbf.predict_active(pred_df=pred_df, k=k)

data/dict_of_all_ids.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3eb6d11b622023bcdaa5a7130886f2d5e5304b0d3764a9a896744b9d34812ae8
+size 820978

data/dict_of_all_titles.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2867960e313b95dae1d80261560df9c8dd63801d49c26fb23fef075b0ee0679c
+size 820978

data/dict_of_ease_ids.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dd9e84631b0dd97fe7d0fe982a28a03c9c65a4d1042437385137d05c92f829b3
+size 60873

data/dict_of_ease_titles.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2689a911254553451cdd573c751192f098d12b3b1cc9d872f7ac2e7b0d840018
+size 60873

data/list_of_all_titles.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4536a375d8a42380eb9501eebab0cbf1a4c17d93c0a88093cba3bb2ab27e3db5
+size 687996

data/list_of_ease_titles.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c99d13e858bb785686cbaeab014d6bb39608627316a6b6b69d5517f21b3fc5ee
+size 50223

func.py

@@ -5,8 +5,8 @@ import requests
 from io import BytesIO
 from PIL import Image
 
-data_titles = pd.read_pickle("data/2k_titles.pkl")
-data_ids = pd.read_pickle("data/2k_ids.pkl")
+data_titles = pd.read_pickle("data/dict_of_all_titles.pkl")
+data_ids = pd.read_pickle("data/dict_of_all_ids.pkl")
 
 def ids_to_titles(ids_list):
     return [data_ids[title] for title in ids_list]