From 04a4b08961021cfaf278fd4bf5d91cb691aef651 Mon Sep 17 00:00:00 2001 From: thor-harsh <105957576+thor-harsh@users.noreply.github.com> Date: Sat, 19 Aug 2023 18:48:32 +0530 Subject: [PATCH] Update k_nearest_neighbours.py --- machine_learning/k_nearest_neighbours.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/machine_learning/k_nearest_neighbours.py b/machine_learning/k_nearest_neighbours.py index 2a90cfe59..c5e0e767b 100644 --- a/machine_learning/k_nearest_neighbours.py +++ b/machine_learning/k_nearest_neighbours.py @@ -10,7 +10,8 @@ X = np.array(data["data"]) y = np.array(data["target"]) classes = data["target_names"] -X_train, X_test, y_train, y_test = train_test_split(X, y) +X_train, X_test, y_train, y_test = train_test_split(X, ym,random_state=42) +'''Random state = 42 is set so that the training and test set remains same after 1st run for all def euclidean_distance(a, b):