Update app.py

app.py

@@ -13,7 +13,7 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.15, random
 
 from sklearn.compose import ColumnTransformer
 from sklearn.pipeline import Pipeline
-from sklearn.metrics import mean_absolute_error
+from sklearn.metrics import mean_squared_error
 from sklearn.preprocessing import StandardScaler, OneHotEncoder
 from sklearn.ensemble import HistGradientBoostingRegressor
 hgbr = HistGradientBoostingRegressor()
@@ -25,7 +25,7 @@ pipe = Pipeline(steps=[('preprocessor', transformer), ('model', hgbr)])
 pipe.fit(X_train, y_train)
 score = pipe.score(X_test,y_test)
 y_pred = pipe.predict(X_test)
-mae = mean_absolute_error(y_test,y_pred)
+mse = mean_squared_error(y_test,y_pred)
 import streamlit as st
 
 def price(companyName,modelName,modelYear,locaiton,mileage,engineType,engineCapacity,color,assembly,bodyType,transmissionType,registrationStatus):
@@ -46,7 +46,7 @@ def price(companyName,modelName,modelYear,locaiton,mileage,engineType,engineCapa
   prediction=pipe.predict(input_data)[0]
   return prediction
 st.title('Car Price Prediction :car: :arrow_forward: :dollar: @yusufenes')
-st.success(f'Accuracy : {score.round(3)}  Mean Absolute Error : {mae.round(2)}')
+st.success(f'Accuracy : {score.round(3)}  MSE : {mae.round(2)}')
 st.write('Please Chose Car Specifications')
 companyName = st.selectbox('Company Name',df['Company Name'].unique())
 modelName = st.selectbox('Model Name',df[df['Company Name']==companyName]['Model Name'].unique())