setuagrawal commited on
Commit
cdd4353
·
verified ·
1 Parent(s): 3e31567

Create preprocessors.py

Browse files
Files changed (1) hide show
  1. preprocessors.py +19 -0
preprocessors.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # preprocessors.py
2
+ import pandas as pd
3
+
4
+ def add_store_age(df):
5
+ df = df.copy()
6
+ df['Store_Age'] = 2025 - df['Store_Established_Year']
7
+ df = df.drop(columns=['Store_Established_Year'])
8
+ return df
9
+
10
+ def map_ordered_features(X):
11
+ sugar_order_map = {'No Sugar': 0, 'Low Sugar': 1, 'Regular': 2}
12
+ size_order_map = {'Small': 0, 'Medium': 1, 'High': 2}
13
+ city_order_map = {'Tier 3': 0, 'Tier 2': 1, 'Tier 1': 2}
14
+
15
+ X = X.copy()
16
+ X['Product_Sugar_Content'] = X['Product_Sugar_Content'].map(sugar_order_map).astype(int)
17
+ X['Store_Size'] = X['Store_Size'].map(size_order_map).astype(int)
18
+ X['Store_Location_City_Type'] = X['Store_Location_City_Type'].map(city_order_map).astype(int)
19
+ return X