Spaces:
Paused
Paused
Update app.py
Browse files
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())
|