commit 24/11/22

eda.py

@@ -27,6 +27,15 @@ def run():
 
     st.subheader('Area Income')
 
+    eda = df.copy()
+    eda['Pricerange'] = 'Very High'
+    eda.loc[eda['Price'].between(1.5e+06, 1.75e+06), 'Pricerange'] = 'High'
+    eda.loc[eda['Price'].between(1.25e+06, 1.5e+06), 'Pricerange'] = 'Above Average'
+    eda.loc[eda['Price'].between(1e+06, 1.25e+06), 'Pricerange'] = 'Below Average'
+    eda.loc[eda['Price'].between(0.75e+06, 1e+06), 'Pricerange'] = 'Low'
+    eda.loc[eda['Price']<0.75e+06, 'Pricerange'] = 'Very Low'
+    st.dataframe(eda)
+
     def barc(df, x, y):
         plot = df.groupby(x).mean().reset_index()
         plot = plot.sort_values(y)
@@ -36,14 +45,6 @@ def run():
         plt.ylabel('Mean')
         st.pyplot(fig)
 
-    eda = df.copy()
-    eda['Pricerange'] = 'Very High'
-    eda.loc[eda['Price'].between(1.5e+06, 1.75e+06), 'Pricerange'] = 'High'
-    eda.loc[eda['Price'].between(1.25e+06, 1.5e+06), 'Pricerange'] = 'Above Average'
-    eda.loc[eda['Price'].between(1e+06, 1.25e+06), 'Pricerange'] = 'Below Average'
-    eda.loc[eda['Price'].between(0.75e+06, 1e+06), 'Pricerange'] = 'Low'
-    eda.loc[eda['Price']<0.75e+06, 'Pricerange'] = 'Very Low'
-
     barc(eda,'Pricerange','Income')
 
     st.markdown('---')

prediction.py

@@ -9,8 +9,8 @@ import joblib
 import json
 
 # Load All Files
-with open('model_lin_reg.pkl', 'rb') as file_1:
-  model_lin_reg = joblib.load(file_1)
+with open('model.pkl', 'rb') as file_1:
+  model = joblib.load(file_1)
 
 with open('pipeline.pkl', 'rb') as file_2:
   preprocessor = joblib.load(file_2)
@@ -57,7 +57,7 @@ def run():
         X_inf = preprocessor.transform(data_inf)
 
         # Predict using Linear regression
-        y_pred_inf = model_lin_reg.predict(X_inf)
+        y_pred_inf = model.predict(X_inf)
 
         st.write('# House Price : ', str(int(y_pred_inf)))