Spaces:
Runtime error
Runtime error
Upload 5 files
Browse files- food_app.py +123 -0
- main.py +186 -0
- model.pickle +3 -0
- predict.py +18 -0
- 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
|