commit

app.py

@@ -34,24 +34,16 @@ else:
 
 knn = KNeighborsRegressor(n_neighbors=K, metric=metric)
 knn.fit(x_train, y_train)
-
+y_pred = knn.predict(x_test)
 
 plt.figure()
 
-x = np.linspace(np.min(x_test), np.max(x_test), 70)
-
-xx = np.meshgrid(x)
-xy = np.c_[np.array(xx).ravel()]
-y_pred = knn.predict(xy)
-
-#plt.pcolormesh(y_pred, cmap='jet', alpha=0.2)
-
-#plt.plot(y_test[:30], "C0s", label="True Points (Test)")
+plt.plot(y_test[:30], "C0s", label="True Points (Test)")
 plt.plot(y_pred[:30], "C1*", label="Predictions (Test)")
 plt.xlabel("X")
 plt.ylabel("Y")
 plt.legend(loc="upper left")
-#plt.ylim(-90,90)
+plt.ylim(-90,90)
 
 sns.despine(right=True, top=True)