Spaces:
Sleeping
Sleeping
Create preprocessors.py
Browse files- 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
|