From 65be793beb970669e48bcf81a05dd547eb80f943 Mon Sep 17 00:00:00 2001
From: lucile <lucile.audard@ecl20.ec-lyon.fr>
Date: Fri, 10 Nov 2023 16:27:26 +0100
Subject: [PATCH] Update knn.py
---
knn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/knn.py b/knn.py
index a7decbb..8b7450e 100644
--- a/knn.py
+++ b/knn.py
@@ -49,7 +49,7 @@ if __name__ == "__main__":
data, labels = read_cifar("./data/cifar-10-batches-py")
# Formatting the data into training and testing sets
- data_train, labels_train, data_test, labels_test = split_dataset(data, labels, 0.1)
+ data_train, labels_train, data_test, labels_test = split_dataset(data, labels, 0.9)
# Data to plot
k_list = [k for k in range(1, 21)]
--
GitLab