Update app.py

app.py

@@ -30,6 +30,8 @@ def homework01_solution1(K, X1, X2):
 
   train_data['Predicted Label ('+str(K)+'-NN)'] = pred
   train_data['Squaredd Error'] = squared_error
+  train_data['Predicted Label ('+str(K)+'-NN)'] = train_data['Predicted Label ('+str(K)+'-NN)'].round(3)
+  train_data['Squaredd Error'] = train_data['Squaredd Error'].round(3)
 
   
     
@@ -58,7 +60,7 @@ def homework01_solution1(K, X1, X2):
     results = pd.concat([results, tmp], ignore_index=True)
 
   results = results.sort_values(by='Rank of closest neighbor')
-  results
+  results['Distance to query data'] = results['Distance to query data'].round(3)
   
   return train_data, results, predicted_label