Update data_preparation.py

data_preparation.py

@@ -1,7 +1,7 @@
 import pandas as pd
 import numpy as np
 def data_imp():
-    feature_descriptions = {
+    insurance_feature_descriptions = {
     "CustID": "Unique identifier for each customer.",
     "FirstPolYear": "Year when the customer first bought an insurance policy.",
     "BirthYear": "Birth year of the customer, used to calculate age.",
@@ -17,6 +17,37 @@ def data_imp():
     "PremLife": "Premium amount for life insurance.",
     "PremWork": "Premium amount for work insurance."
     }
+    retail_feature_descriptions = {
+    "Channel": "Indicates the sales channel through which the customer made purchases.",
+    "Region": "The geographical region where the customer is located.",
+    "Fresh": "Annual spending (in monetary units) on fresh products.",
+    "Milk": "Annual spending (in monetary units) on milk products.",
+    "Grocery": "Annual spending (in monetary units) on grocery items.",
+    "Frozen": "Annual spending (in monetary units) on frozen products.",
+    "Detergents_Paper": "Annual spending (in monetary units) on detergents and paper products.",
+    "Delicassen": "Annual spending (in monetary units) on delicatessen products."
+    }
+    bankng_feature_descriptions = {
+    "CUST_ID": "Unique identifier for each customer.",
+    "BALANCE": "The average balance left in the customer's account.",
+    "BALANCE_FREQUENCY": "Frequency with which the balance is updated.",
+    "PURCHASES": "The total amount of purchases made by the customer.",
+    "ONEOFF_PURCHASES": "The total amount of one-time purchases made by the customer.",
+    "INSTALLMENTS_PURCHASES": "The total amount of purchases made in installments.",
+    "CASH_ADVANCE": "The total amount of cash advances taken by the customer.",
+    "PURCHASES_FREQUENCY": "The frequency of purchases made by the customer.",
+    "ONEOFF_PURCHASES_FREQUENCY": "The frequency of one-time purchases made by the customer.",
+    "PURCHASES_INSTALLMENTS_FREQUENCY": "The frequency of purchases made in installments.",
+    "CASH_ADVANCE_FREQUENCY": "The frequency of cash advances taken by the customer.",
+    "CASH_ADVANCE_TRX": "The number of cash advance transactions made by the customer.",
+    "PURCHASES_TRX": "The number of purchase transactions made by the customer.",
+    "CREDIT_LIMIT": "The credit limit assigned to the customer's account.",
+    "PAYMENTS": "The total amount of payments made by the customer.",
+    "MINIMUM_PAYMENTS": "The minimum amount of payments made by the customer.",
+    "PRC_FULL_PAYMENT": "The percentage of full payments made by the customer.",
+    "TENURE": "The tenure of the customer in months."
+    }
+
     insurance_defaults = {
                 "FirstPolYear": 1999,
                 "BirthYear": 1980,
@@ -62,7 +93,7 @@ def data_imp():
                 "Detergents_Paper": 4000,
                 "Delicassen": 2000
             }
-    return feature_descriptions,insurance_defaults,banking_defaults,retail_defaults
+    return insurance_feature_descriptions,bankng_feature_descriptions,retail_feature_descriptions,insurance_defaults,banking_defaults,retail_defaults
 
 def preprocess_data(data):
     if 'CustID' in data.columns: