Update utils/data_processing.py

utils/data_processing.py

@@ -54,6 +54,17 @@ def get_data(dataset_name, different_new_nodes_between_val_and_test=False, rando
   graph_df = pd.read_csv('./data/ml_{}.csv'.format(dataset_name))
   edge_features = np.load('./data/ml_{}.npy'.format(dataset_name))
   node_features = np.load('./data/ml_{}_node.npy'.format(dataset_name)) 
+
+
+  #ASHUTOSH
+  graph_df = graph_df.rename(columns={
+    'user_id': 'u',
+    'item_id': 'i',
+    'ts': 'ts',
+    'state_label': 'label',
+    # You can drop or handle the features column separately if needed
+  })
+  ###################
     
   if randomize_features:
     node_features = np.random.rand(node_features.shape[0], node_features.shape[1])