-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathKNN.py
More file actions
71 lines (43 loc) · 1.76 KB
/
KNN.py
File metadata and controls
71 lines (43 loc) · 1.76 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import numpy as np
from collections import Counter
####Sample Data
x_val= np.array([6,8,3,1,5,7,9,7,4,4,6,8,3,1])
y_val = np.array([6,4,2,6,7,8,3,2,6,7,8,5,2,1])
labels = ["blue", "green", "red", "purple", "blue", "green", "red", "purple",
"blue", "green", "red", "purple", "red", "purple"]
def manhattan_distance(x,x_test,y,y_test):
xdist = np.sum(np.abs(x - x_test))
ydist =np.sum(np.abs(y - y_test))
return xdist+ydist
def simple_knn(x_test,y_test,x,y,labels):
dist = []
for x,y in zip(x,y):
dist.append(manhattan_distance(x,x_test,y,y_test))
sorted_dist = sorted(dist)
k = 2
pred_label = {}
for index, distance in sorted(zip(range(len(dist)), dist), key=lambda x: x[1])[:k]:
label = labels[index]
pred_label[label] = pred_label.get(label, 0) + 1
pred_label = max(pred_label, key=pred_label.get)
print(pred_label)
return pred_label
simple_knn(4,1,x_val,y_val,labels)
import numpy as np
from sklearn.neighbors import KNeighborsClassifier
# Provided data
x_val = np.array([6, 8, 3, 1, 5, 7, 9, 7, 4, 4, 6, 8, 3, 1])
y_val = np.array([6, 4, 2, 6, 7, 8, 3, 2, 6, 7, 8, 5, 2, 1])
labels = ["blue", "green", "red", "purple", "blue", "green", "red", "purple",
"blue", "green", "red", "purple", "red", "purple"]
# Combine x_val and y_val into feature vectors
feature_vectors = np.column_stack((x_val, y_val))
# Initialize the KNN classifier with k=3
knn_classifier = KNeighborsClassifier(n_neighbors=2)
# Train the classifier
knn_classifier.fit(feature_vectors, labels)
# New data point to predict
new_data_point = np.array([[4, 1]])
# Predict the label for the new data point
predicted_label = knn_classifier.predict(new_data_point)
print("Predicted label:", predicted_label[0])