Update utils.py

utils.py

@@ -47,11 +47,11 @@ def load_and_preprocess_data():
 
     # Create a categorical type for users and product. User ordered to ensure
     # reproducibility
-    #user_cat = pd.CategoricalDtype(categories=sorted(users), ordered=True)
+    user_cat = pd.CategoricalDtype(categories=sorted(users), ordered=True)
     product_cat = pd.CategoricalDtype(categories=sorted(products), ordered=True)
 
     # Transform and get the indexes of the columns
-    user_idx = df["User-ID"]
+    user_idx = df["User-ID"].astype(product_cat).cat.codes
     product_idx = df["ISBN"].astype(product_cat).cat.codes
 
     # Add the categorical index to the starting dataframe
@@ -61,6 +61,7 @@ def load_and_preprocess_data():
     merged_df = pd.merge(df, df_users[['User-ID', 'Location', 'Age']], on='User-ID', how='left')
     merged_df = pd.merge(merged_df, df_books[['ISBN', 'Book-Title', 'Book-Author', 'Year-Of-Publication']], on='ISBN', how='left')
     merged_df["ProductIndex"] = product_idx
+    merged_df["CustomerIndex"] = user_idx