Update driverpospredictionfinal.py

driverpospredictionfinal.py

@@ -184,69 +184,18 @@ tuned_model_3 = hyper_paramter_tuning(model_3,xtrain,ytrain,**parameter_grid)
 y_pred= predict(tuned_model_3,xtest,ytest)
 
 
-# Ask user for input and predict position
-#predicted_position = ask_user(tuned_model_3, x_columns)
-
-
+with open('model.pkl', 'wb') as file:
+    pickle.dump(tuned_model_3, file)
 
 
-predicted_position = ask_user(tuned_model_3, xtest)
-
-#pickle.dump(tuned_model_3.open('model.pkl','mb'))
-
-!pip install streamlit
-import streamlit as st
-import pickle
-import numpy as np
 
-model = pickle.load(open('model.pkl','rb'))
-
-def predictPos(TLaps,Laptime,Pit,Prev,AvgSpeed,AirTemp,TrackTemp,Humid,Wind):
-  input = np.array([ [TLaps,Laptime,Pit,Prev,AvgSpeed,AirTemp,TrackTemp,Humid,Wind]]).astype(np.float64)
-  prediction=model.predict(input)
-  return prediction[0]
-
-def main():
-  st.title("title")
-  html_temp="""
-  <div style = "background-color:#025244; padding :10px">
-  <h2 style = "color:white; text-align:center;"> Driver Position Prediction </h2>
-  </div>
-  """
-  st.markdown(html_temp, unsafe_allow_html=True)
-  TLaps = st.text_input("Enter the number of total laps in race: ","Type Here")
-  Laptime = st.text_input("Avg lap time from prev year (or estimate 98): ","Type Here")
-  Pit = st.text_input("Enter pit stop time (0 if no pit stop): ","Type Here")
-  Prev= st.text_input("Enter avg lap time differece (0-2) (0 if first lap): ","Type Here")
-
-
-  AvgSpeed = st.text_input("Enter average speed in km/h from prev year: ","Type Here")
-  AirTemp= st.text_input("estimate air temperature in Celsius on race day: ","Type Here")
-  TrackTemp = st.text_input("estimate track temperature in Celsius on race day: ","Type Here")
-  Humid = st.text_input("estimate the humidity percentage (0-100) on race day: ","Type Here")
-  Wind = st.text_input("estimate wind speed in km/h: ","Type Here")
-  safe_html="""
-    <div style="background-color:#F4D03F;padding:10px >
-    <h2 style="color:white;text-align:center;"> Your forest is safe</h2>
-    </div>
-    """
-  danger_html="""
-  <div style="background-color:#F08080;padding:10px >
-    <h2 style="color:black ;text-align:center;"> Your forest is in danger</h2>
-   </div>
-    """
+# Ask user for input and predict position
+#predicted_position = ask_user(tuned_model_3, x_columns)
 
-  if st.button("Predict"):
-    output=predictPos(TLaps,Laptime,Pit,Prev,AvgSpeed,AirTemp,TrackTemp,Humid,Wind)
-    st.success('The postion of driver is {}'.format(output))
 
-    # if output > 0.5:
-    #   st.markdown(danger_html,unsafe_allow_html=True)
-    # else:
-    #   st.markdown(safe_html,unsafe_allow_html=True)
 
-if __name__=='__main__':
-    main()
 
+#predicted_position = ask_user(tuned_model_3, xtest)
 
+#pickle.dump(tuned_model_3.open('model.pkl','mb'))