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