azeemkhan417 commited on
Commit
89bb8f5
·
verified ·
1 Parent(s): e9a4a47

Upload 5 files

Browse files
Files changed (5) hide show
  1. food_app.py +123 -0
  2. main.py +186 -0
  3. model.pickle +3 -0
  4. predict.py +18 -0
  5. requirements.txt +3 -0
food_app.py ADDED
@@ -0,0 +1,123 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from pathlib import Path
2
+ import datetime
3
+
4
+ import pandas as pd
5
+ import streamlit as st
6
+
7
+ import main
8
+ import predict
9
+
10
+
11
+ def get_user_input(df_train):
12
+ st.sidebar.write(f"**Order Related Information**")
13
+ date = st.sidebar.date_input("what is the Order Date?")
14
+ order_time = st.sidebar.time_input("What is the Order Time?", step=60)
15
+ order_datetime = datetime.datetime.combine(date, order_time)
16
+ pickup_time = st.sidebar.time_input("What is the Order Pickup Time?",
17
+ order_datetime + datetime.timedelta(minutes=15), step=60)
18
+ order_type = st.sidebar.selectbox('What is the type of order?',
19
+ df_train['Type_of_order'].unique())
20
+ multiple_deliveries = st.sidebar.selectbox('How many deliveries are combined?',
21
+ sorted(df_train['multiple_deliveries'].unique().astype('int')))
22
+ st.sidebar.write(f"**Location Related Information**")
23
+ restaurant_latitude = st.sidebar.text_input("What is the restaurant latitude?", "14.829222")
24
+ restaurant_longitude = st.sidebar.text_input("What is the restaurant longitude?", "67.920922")
25
+ delivery_location_latitude = st.sidebar.text_input("What is the delivery location latitude?", "14.929222")
26
+ delivery_location_longitude = st.sidebar.text_input("What is the delivery location longitude?", "68.860922")
27
+ st.sidebar.write(f"**Delivery Person Related Information**")
28
+ delivery_person_age = st.sidebar.slider("How old is the delivery person?",
29
+ int(df_train['Delivery_person_Age'].min()),
30
+ int(df_train['Delivery_person_Age'].max()),
31
+ int(df_train['Delivery_person_Age'].mean()))
32
+ delivery_person_rating = st.sidebar.slider("What is delivery person rating?",
33
+ float(df_train['Delivery_person_Ratings'].min()),
34
+ float(df_train['Delivery_person_Ratings'].max()),
35
+ float(df_train['Delivery_person_Ratings'].mean()))
36
+ vehicle = st.sidebar.selectbox('What type of vehicle delivery person has?',
37
+ df_train['Type_of_vehicle'].unique())
38
+ vehicle_condition = st.sidebar.selectbox('What is the Vehicle condition of delivery person?',
39
+ sorted(df_train['Vehicle_condition'].unique()))
40
+ st.sidebar.write(f"**City Related Information**")
41
+ city_code = st.sidebar.selectbox('What is the city name of delivery?',
42
+ df_train['City_code'].unique())
43
+ city = st.sidebar.selectbox('Which type of city it is?',
44
+ df_train['City'].unique())
45
+ st.sidebar.write(f"**Weather Conditions/Event Related Information**")
46
+ road_density = st.sidebar.selectbox('What is road traffic density?',
47
+ df_train['Road_traffic_density'].unique())
48
+ weather_conditions = st.sidebar.selectbox('How is the weather?',
49
+ df_train['Weather_conditions'].unique())
50
+ festival = st.sidebar.selectbox('Is there a festival?',
51
+ df_train['Festival'].unique())
52
+
53
+ X = pd.DataFrame({
54
+ 'ID': '123456',
55
+ 'Delivery_person_ID': city_code + 'RES13DEL02',
56
+ 'Delivery_person_Age': delivery_person_age,
57
+ 'Delivery_person_Ratings': delivery_person_rating,
58
+ 'Restaurant_latitude': format(float(restaurant_latitude), ".6f"),
59
+ 'Restaurant_longitude': format(float(restaurant_longitude), ".6f"),
60
+ 'Delivery_location_latitude': format(float(delivery_location_latitude), ".6f"),
61
+ 'Delivery_location_longitude': format(float(delivery_location_longitude), ".6f"),
62
+ 'Order_Date': date.strftime('%d-%m-%Y'),
63
+ 'Time_Orderd': order_time.strftime('%H:%M:%S'),
64
+ 'Time_Order_picked': pickup_time.strftime('%H:%M:%S'),
65
+ 'Weatherconditions': 'conditions ' + weather_conditions,
66
+ 'Road_traffic_density': road_density,
67
+ 'Vehicle_condition': vehicle_condition,
68
+ 'Type_of_order': order_type,
69
+ 'Type_of_vehicle': vehicle,
70
+ 'multiple_deliveries': multiple_deliveries,
71
+ 'Festival': festival,
72
+ 'City': city
73
+ }, index=[0])
74
+ return X
75
+
76
+
77
+ if __name__ == "__main__":
78
+ st.set_page_config(page_title="Food Delivery Time Prediction", page_icon=None, layout="centered",
79
+ initial_sidebar_state="auto")
80
+
81
+ # Read in training data
82
+ df_train = pd.read_csv(str(Path(__file__).parents[1] / 'data/train.csv'))
83
+ main.cleaning_steps(df_train)
84
+
85
+ # Displaying text
86
+ st.title("Food Delivery Time Prediction")
87
+ # Displaying an image
88
+ st.image(str(Path(__file__).parents[1] / 'img/food-delivery.png'), width=700)
89
+
90
+ st.write("""
91
+ The food delivery time prediction model is vital in ensuring prompt and accurate delivery in the food delivery industry. Leveraging advanced data cleaning techniques and feature engineering, a robust food delivery time prediction model is developed.
92
+
93
+ This model predicts food delivery time based on a range of factors, including order details, location, city, delivery person information, and weather conditions.
94
+ """)
95
+
96
+ ##create the sidebar
97
+ st.sidebar.header("User Input Parameters")
98
+
99
+ ##create function for User input
100
+
101
+ input_df = get_user_input(df_train) # get user input from sidebar
102
+
103
+ order_date = input_df['Order_Date'][0]
104
+ order_time = input_df['Time_Orderd'][0]
105
+ order_date_time = datetime.datetime.strptime(f'{order_date} {order_time}', '%d-%m-%Y %H:%M:%S')
106
+ order_pickup_time = input_df['Time_Order_picked'][0]
107
+ order_pickup_date_time = datetime.datetime.strptime(f'{order_date} {order_pickup_time}', '%d-%m-%Y %H:%M:%S')
108
+
109
+ total_delivery_minutes = round(predict.predict(input_df)[0], 2) # get predicitions
110
+ minutes = int(total_delivery_minutes)
111
+ seconds = int((total_delivery_minutes - minutes) * 60)
112
+ X = order_pickup_date_time + datetime.timedelta(minutes=minutes, seconds=seconds)
113
+
114
+ # display predictions
115
+ st.subheader("Order Details")
116
+ st.write(f"**Order was Placed on :** {order_date_time}")
117
+ st.write(f"**Order was Picked up at :** {order_pickup_date_time}")
118
+ st.subheader("Prediction")
119
+ formatted_X = "{:.2f}".format(total_delivery_minutes)
120
+ st.write(f"**Total Delivery Time is :** {formatted_X} mins")
121
+ st.write(f"**Order will be delivered by :** {X}")
122
+
123
+
main.py ADDED
@@ -0,0 +1,186 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from pathlib import Path
2
+ import numpy as np
3
+ import pandas as pd
4
+ from geopy.distance import geodesic
5
+ import pickle
6
+ from sklearn.model_selection import train_test_split
7
+ from sklearn.preprocessing import LabelEncoder, StandardScaler
8
+ import xgboost as xgb
9
+ from sklearn.metrics import mean_squared_error, r2_score, mean_absolute_error
10
+
11
+
12
+ def update_column_name(df):
13
+ df.rename(columns={'Weatherconditions': 'Weather_conditions'}, inplace=True)
14
+
15
+
16
+ def extract_feature_value(df):
17
+ # Extract Weather conditions
18
+ df['Weather_conditions'] = df['Weather_conditions'].apply(lambda x: x.split(' ')[1].strip())
19
+ # Extract city code from Delivery person ID
20
+ df['City_code'] = df['Delivery_person_ID'].str.split("RES", expand=True)[0]
21
+
22
+ #Remove Whitespaces on categorical value
23
+ categorical_columns = df.select_dtypes(include='object').columns
24
+ for column in categorical_columns:
25
+ df[column] = df[column].str.strip()
26
+
27
+
28
+ def extract_label_value(df):
29
+ # Extract time and convert to int
30
+ df['Time_taken(min)'] = df['Time_taken(min)'].apply(lambda x: int(x.split(' ')[1].strip()))
31
+
32
+
33
+ def drop_columns(df):
34
+ df.drop(['ID', 'Delivery_person_ID'], axis=1, inplace=True)
35
+
36
+
37
+ def update_datatype(df):
38
+ df['Delivery_person_Age'] = df['Delivery_person_Age'].astype('float64')
39
+ df['Delivery_person_Ratings'] = df['Delivery_person_Ratings'].astype('float64')
40
+ df['multiple_deliveries'] = df['multiple_deliveries'].astype('float64')
41
+ df['Order_Date'] = pd.to_datetime(df['Order_Date'], format="%d-%m-%Y")
42
+
43
+
44
+ def convert_nan(df):
45
+ df.replace('NaN', float(np.nan), regex=True, inplace=True)
46
+
47
+
48
+ def handle_null_values(df):
49
+ df['Delivery_person_Age'].fillna(np.random.choice(df['Delivery_person_Age']), inplace=True)
50
+ df['Weather_conditions'].fillna(np.random.choice(df['Weather_conditions']), inplace=True)
51
+ df['City'].fillna(df['City'].mode()[0], inplace=True)
52
+ df['Festival'].fillna(df['Festival'].mode()[0], inplace=True)
53
+ df['multiple_deliveries'].fillna(df['multiple_deliveries'].mode()[0], inplace=True)
54
+ df['Road_traffic_density'].fillna(df['Road_traffic_density'].mode()[0], inplace=True)
55
+ df['Delivery_person_Ratings'].fillna(df['Delivery_person_Ratings'].median(), inplace=True)
56
+
57
+
58
+ def extract_date_features(data):
59
+ data["day"] = data.Order_Date.dt.day
60
+ data["month"] = data.Order_Date.dt.month
61
+ data["quarter"] = data.Order_Date.dt.quarter
62
+ data["year"] = data.Order_Date.dt.year
63
+ data['day_of_week'] = data.Order_Date.dt.day_of_week.astype(int)
64
+ data["is_month_start"] = data.Order_Date.dt.is_month_start.astype(int)
65
+ data["is_month_end"] = data.Order_Date.dt.is_month_end.astype(int)
66
+ data["is_quarter_start"] = data.Order_Date.dt.is_quarter_start.astype(int)
67
+ data["is_quarter_end"] = data.Order_Date.dt.is_quarter_end.astype(int)
68
+ data["is_year_start"] = data.Order_Date.dt.is_year_start.astype(int)
69
+ data["is_year_end"] = data.Order_Date.dt.is_year_end.astype(int)
70
+ data['is_weekend'] = np.where(data['day_of_week'].isin([5, 6]), 1, 0)
71
+
72
+
73
+ def calculate_time_diff(df):
74
+ # Find the difference between ordered time & picked time
75
+ df['Time_Orderd'] = pd.to_timedelta(df['Time_Orderd'])
76
+ df['Time_Order_picked'] = pd.to_timedelta(df['Time_Order_picked'])
77
+
78
+ df['Time_Order_picked_formatted'] = df['Order_Date'] + np.where(df['Time_Order_picked'] < df['Time_Orderd'],
79
+ pd.DateOffset(days=1), pd.DateOffset(days=0)) + df[
80
+ 'Time_Order_picked']
81
+ df['Time_Ordered_formatted'] = df['Order_Date'] + df['Time_Orderd']
82
+
83
+ df['order_prepare_time'] = (df['Time_Order_picked_formatted'] - df[
84
+ 'Time_Ordered_formatted']).dt.total_seconds() / 60
85
+
86
+ # Handle null values by filling with the median
87
+ df['order_prepare_time'].fillna(df['order_prepare_time'].median(), inplace=True)
88
+
89
+ # Drop all the time & date related columns
90
+ df.drop(['Time_Orderd', 'Time_Order_picked', 'Time_Ordered_formatted', 'Time_Order_picked_formatted', 'Order_Date'],
91
+ axis=1, inplace=True)
92
+
93
+
94
+ def calculate_distance(df):
95
+ df['distance'] = np.zeros(len(df))
96
+ restaurant_coordinates = df[['Restaurant_latitude', 'Restaurant_longitude']].to_numpy()
97
+ delivery_location_coordinates = df[['Delivery_location_latitude', 'Delivery_location_longitude']].to_numpy()
98
+ df['distance'] = np.array([geodesic(restaurant, delivery) for restaurant, delivery in
99
+ zip(restaurant_coordinates, delivery_location_coordinates)])
100
+ df['distance'] = df['distance'].astype("str").str.extract('(\d+)').astype("int64")
101
+
102
+
103
+ def label_encoding(df):
104
+ categorical_columns = df.select_dtypes(include='object').columns
105
+ label_encoders = {}
106
+
107
+ # Iterate over each categorical column and fit a label encoder
108
+ for column in categorical_columns:
109
+ df[column] = df[column].str.strip() # Remove whitespaces
110
+ label_encoder = LabelEncoder()
111
+ label_encoder.fit(df[column])
112
+ df[column] = label_encoder.transform(df[column])
113
+ label_encoders[column] = label_encoder
114
+ return label_encoders
115
+
116
+
117
+ def data_split(X, y):
118
+ # Split the data into train and test sets
119
+ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
120
+
121
+ return X_train, X_test, y_train, y_test
122
+
123
+
124
+ def standardize(X_train, X_test):
125
+ scaler = StandardScaler()
126
+
127
+ # Fit the scaler on the training data
128
+ scaler.fit(X_train)
129
+
130
+ # Perform standardization
131
+ X_train = scaler.transform(X_train)
132
+ X_test = scaler.transform(X_test)
133
+ return X_train, X_test, scaler
134
+
135
+
136
+ def cleaning_steps(df):
137
+ update_column_name(df)
138
+ extract_feature_value(df)
139
+ drop_columns(df)
140
+ update_datatype(df)
141
+ convert_nan(df)
142
+ handle_null_values(df)
143
+
144
+ def perform_feature_engineering(df):
145
+ extract_date_features(df)
146
+ calculate_time_diff(df)
147
+ calculate_distance(df)
148
+
149
+
150
+ def evaluate_model(y_test, y_pred):
151
+ mae = mean_absolute_error(y_test, y_pred)
152
+ mse = mean_squared_error(y_test, y_pred)
153
+ rmse = np.sqrt(mse)
154
+ r2 = r2_score(y_test, y_pred)
155
+
156
+ print("Mean Absolute Error (MAE):", round(mae, 2))
157
+ print("Mean Squared Error (MSE):", round(mse, 2))
158
+ print("Root Mean Squared Error (RMSE):", round(rmse, 2))
159
+ print("R-squared (R2) Score:", round(r2, 2))
160
+
161
+
162
+ if __name__ == "__main__":
163
+ df_train = pd.read_csv(str(Path(__file__).parents[1] / 'data/train.csv')) # Load Data
164
+ cleaning_steps(df_train) # Perform Cleaning
165
+ extract_label_value(df_train) #Extract Label Value
166
+ perform_feature_engineering(df_train) # Perform feature engineering
167
+
168
+ # Split features & label
169
+ X = df_train.drop('Time_taken(min)', axis=1) # Features
170
+ y = df_train['Time_taken(min)'] # Target variable
171
+
172
+ label_encoders = label_encoding(X) # Label Encoding
173
+ X_train, X_test, y_train, y_test = data_split(X, y) # Test Train Split
174
+ X_train, X_test, scaler = standardize(X_train, X_test) # Standardization
175
+
176
+ # Build Model
177
+ model = xgb.XGBRegressor(n_estimators=20, max_depth=9)
178
+ model.fit(X_train, y_train)
179
+
180
+ # Evaluate Model
181
+ y_pred = model.predict(X_test)
182
+ evaluate_model(y_test, y_pred)
183
+
184
+ # Save Model
185
+ with open(str(Path(__file__).parents[1] / 'code/model.pickle'), 'wb') as f:
186
+ pickle.dump((model, label_encoders, scaler), f)
model.pickle ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcb220c00ae0f583df1b423f37ea04d0691676ffe17ca9dbd608eba0dba3c4d0
3
+ size 373904
predict.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from pathlib import Path
2
+ import main
3
+ import pickle
4
+
5
+
6
+ def predict(X):
7
+ # Load the model and scaler from the saved file
8
+ with open(str(Path(__file__).parents[1] / 'code/model.pickle'), 'rb') as f:
9
+ model, label_encoders, scaler = pickle.load(f)
10
+
11
+ main.cleaning_steps(X) # Perform Cleaning
12
+ main.perform_feature_engineering(X) # Perform Feature Engineering
13
+ # Label Encoding
14
+ for column, label_encoder in label_encoders.items():
15
+ X[column] = label_encoder.transform(X[column])
16
+ X = scaler.transform(X) # Standardize
17
+ pred = model.predict(X)
18
+ return pred
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ geopy==2.3.0
2
+ xgboost==1.6.2
3
+ scikit-learn==1.2.1