YoneSlapWind80085 commited on
Commit
b5aaf69
·
verified ·
1 Parent(s): 486f5eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -203,8 +203,9 @@ ax.set_ylabel(selected_target)
203
 
204
  # Show regression line if selected
205
  show_regression = st.checkbox('Show Regression Line')
206
- if show_regression and selected_feature in X.columns and selected_target == 'MedHouseVal':
207
  X_feature = df[[selected_feature]]
 
208
  model_feature = LinearRegression()
209
  model_feature.fit(X_feature, y)
210
  line = model_feature.predict(X_feature)
 
203
 
204
  # Show regression line if selected
205
  show_regression = st.checkbox('Show Regression Line')
206
+ if show_regression and selected_feature in df.columns and selected_target == 'MedHouseVal':
207
  X_feature = df[[selected_feature]]
208
+ y = df[selected_target]
209
  model_feature = LinearRegression()
210
  model_feature.fit(X_feature, y)
211
  line = model_feature.predict(X_feature)