Update app.py

app.py

@@ -19,6 +19,12 @@ y_test = y[ntrain:]
 knn = KNN(n_neighbors=K)
 knn.fit(x_train, y_train)
 plt.figure()
+
+x = np.linspace(np.min(x_test[:, 0]), np.max(x_test[:, 0]), 200)
+y = np.linspace(np.min(x_test[:, 1]), np.max(x_test[:, 1]), 200)
+xx, yy = np.meshgrid(x, y)
+xy = np.c_[xx.ravel(), yy.ravel()]
+
 y_predicted = knn.predict(xy)
 #plt.pcolormesh(y_predicted.reshape(200, 200), cmap='jet')
 plt.pcolormesh(xx, yy, y_predicted.reshape(200, 200), cmap='jet', alpha=0.2)