Emil25 commited on
Commit
4802bdb
·
verified ·
1 Parent(s): 12c79cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -123,7 +123,7 @@ def get_haversine_distance(lat1, lng1, lat2, lng2):
123
 
124
 
125
  # User input features
126
- def user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count, temperature):
127
  current_time = datetime.now()
128
  pickup_hour= current_time.hour
129
  today = datetime.today()
@@ -145,7 +145,7 @@ def user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count, tem
145
  'total_travel_time': total_travel_time,
146
  'number_of_steps': number_of_steps,
147
  'haversine_distance': haversine_distance,
148
- 'temperature': temperature,
149
  'pickup_day_of_week_1': 1 if weekday_number == 1 else 0,
150
  'pickup_day_of_week_2': 1 if weekday_number == 2 else 0,
151
  'pickup_day_of_week_3': 1 if weekday_number == 3 else 0,
@@ -180,14 +180,14 @@ def main():
180
  address_from = st.sidebar.text_input("Откуда:", value="New York, 11 Wall Street")
181
  address_to = st.sidebar.text_input("Куда:", value="New York, 740 Park Avenue")
182
  passenger_count = st.sidebar.slider("Количество пассажиров", 1, 4, 1)
183
- temperature = st.sidebar.slider("Temperature (C)", -20, 40, 15)
184
  st.session_state['btn_predict'] = st.sidebar.button('Start')
185
 
186
  if st.session_state['btn_predict']:
187
  lon_from, lat_from = get_coordinates(address_from)
188
  lon_to, lat_to = get_coordinates(address_to)
189
  st.plotly_chart(show_map(lon_from, lat_from, lon_to, lat_to))
190
- user_data = user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count, temperature)
191
  # st.write(user_data)
192
  data_scaled = min_max_scaler(user_data)
193
  trip_duration = np.exp(make_prediction(data_scaled)) - 1
 
123
 
124
 
125
  # User input features
126
+ def user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count):
127
  current_time = datetime.now()
128
  pickup_hour= current_time.hour
129
  today = datetime.today()
 
145
  'total_travel_time': total_travel_time,
146
  'number_of_steps': number_of_steps,
147
  'haversine_distance': haversine_distance,
148
+ 'temperature': 15,
149
  'pickup_day_of_week_1': 1 if weekday_number == 1 else 0,
150
  'pickup_day_of_week_2': 1 if weekday_number == 2 else 0,
151
  'pickup_day_of_week_3': 1 if weekday_number == 3 else 0,
 
180
  address_from = st.sidebar.text_input("Откуда:", value="New York, 11 Wall Street")
181
  address_to = st.sidebar.text_input("Куда:", value="New York, 740 Park Avenue")
182
  passenger_count = st.sidebar.slider("Количество пассажиров", 1, 4, 1)
183
+
184
  st.session_state['btn_predict'] = st.sidebar.button('Start')
185
 
186
  if st.session_state['btn_predict']:
187
  lon_from, lat_from = get_coordinates(address_from)
188
  lon_to, lat_to = get_coordinates(address_to)
189
  st.plotly_chart(show_map(lon_from, lat_from, lon_to, lat_to))
190
+ user_data = user_input_features(lon_from, lat_from, lon_to, lat_to, passenger_count)
191
  # st.write(user_data)
192
  data_scaled = min_max_scaler(user_data)
193
  trip_duration = np.exp(make_prediction(data_scaled)) - 1