0_2_Dataset_construction.py

@@ -124,6 +124,9 @@ def dataset_construction(min_date, end_features_date, max_date, directory_path):
     # 0 if there is no purchase
     dataset[['y_has_purchased']].fillna(0)
 
+    # add id_company prefix to customer_id
+    dataset['customer_id'] = directory_path + '_' + dataset['customer_id'].astype('str')
+    
     return dataset
 
 ## Exportation