yusufenes commited on
Commit
4c16457
·
verified ·
1 Parent(s): 7258f72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -13,6 +13,7 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.15, random
13
 
14
  from sklearn.compose import ColumnTransformer
15
  from sklearn.pipeline import Pipeline
 
16
  from sklearn.preprocessing import StandardScaler, OneHotEncoder
17
  from sklearn.ensemble import HistGradientBoostingRegressor
18
  hgbr = HistGradientBoostingRegressor()
@@ -22,7 +23,9 @@ transformer = ColumnTransformer(transformers=[('num', StandardScaler(), num_feat
22
  pipe = Pipeline(steps=[('preprocessor', transformer), ('model', hgbr)])
23
 
24
  pipe.fit(X_train, y_train)
25
-
 
 
26
  import streamlit as st
27
 
28
  def price(companyName,modelName,modelYear,locaiton,mileage,engineType,engineCapacity,color,assembly,bodyType,transmissionType,registrationStatus):
@@ -43,6 +46,7 @@ def price(companyName,modelName,modelYear,locaiton,mileage,engineType,engineCapa
43
  prediction=pipe.predict(input_data)[0]
44
  return prediction
45
  st.title('Car Price Prediction :car: :arrow_forward: :dollar: @yusufenes')
 
46
  st.write('Please Chose Car Specifications')
47
  companyName = st.selectbox('Company Name',df['Company Name'].unique())
48
  modelName = st.selectbox('Model Name',df[df['Company Name']==companyName]['Model Name'].unique())
 
13
 
14
  from sklearn.compose import ColumnTransformer
15
  from sklearn.pipeline import Pipeline
16
+ from sklearn.metrics import mean_absolute_error
17
  from sklearn.preprocessing import StandardScaler, OneHotEncoder
18
  from sklearn.ensemble import HistGradientBoostingRegressor
19
  hgbr = HistGradientBoostingRegressor()
 
23
  pipe = Pipeline(steps=[('preprocessor', transformer), ('model', hgbr)])
24
 
25
  pipe.fit(X_train, y_train)
26
+ score = pipe.score(X_test,y_test)
27
+ y_pred = pipe.predict(X_test)
28
+ mae = mean_absolute_error(y_test,y_pred)
29
  import streamlit as st
30
 
31
  def price(companyName,modelName,modelYear,locaiton,mileage,engineType,engineCapacity,color,assembly,bodyType,transmissionType,registrationStatus):
 
46
  prediction=pipe.predict(input_data)[0]
47
  return prediction
48
  st.title('Car Price Prediction :car: :arrow_forward: :dollar: @yusufenes')
49
+ st.write(f'Accuracy : {score} Mean Absolute Error : {mae}')
50
  st.write('Please Chose Car Specifications')
51
  companyName = st.selectbox('Company Name',df['Company Name'].unique())
52
  modelName = st.selectbox('Model Name',df[df['Company Name']==companyName]['Model Name'].unique())