Update app.py

app.py

@@ -24,6 +24,16 @@ def homework01_solution1(K, X1, X2):
   neigh.fit(X, y)
   predicted_label = neigh.predict(np.array([[X1, X2]]))[0]
 
+  #(Q) calculate squared error
+  pred = neigh.predict(X)
+  squared_error = (pred-y)**2
+
+  train_data['Predicted Label ('+K+'-NN)'] = pred
+  train_data['Squaredd Error'] = squared_error
+
+  
+    
+
   (nb_dist, nb_indice) = neigh.kneighbors(np.array([[X1, X2]]), K)
 
   import pandas as pd