Update app.py

app.py

@@ -17,15 +17,15 @@ def make_data(dataset_option):
     opt = dataset_option.split()[0]
     if opt == "100":
         X, y = make_regression(n_samples=100,
-                               n_features=2,
+                               n_features=10, n_informative=2,
                                random_state=42)
     elif opt == "200":
         X, y = make_regression(n_samples=200,
-                               n_features=2,
+                               n_features=5, n_informative=2,
                                random_state=56)
     elif opt == "150":
         X, y = make_regression(n_samples=150,
-                               n_features=2,
+                               n_features=7, n_informative=2,
                                random_state=25)
     else:
         X, y = make_regression(random_state=10)
@@ -73,8 +73,9 @@ if st.button('Magic'):
     y_pred = best_model.predict(X_test)
     fig = plt.figure()
     plt.title(f"Best model fit is of {opts[np.argmin(loss)]}")
-    plt.scatter(X_test[:,0], y_pred)
-    plt.scatter(X_test[:,0], y_test)
+    plt.scatter(X_test[:,0], y_pred, label = "Prediction Value")
+    plt.scatter(X_test[:,0], y_test, label = "Real Value")
+    plt.legend()
     st.pyplot(fig)
 
 options = ['LinearRegressor', 'DecisionTreeRegressor', 'SVR']