cs231n lecture2 image classification

 1 import numpy as np
 2 
 3 class NearestNeighbor:
 4     def _init_(self):
 5         pass
 6   
 7     def train(self, X, y):
 8         self.Xtr = X
 9         self.ytr = y
10   
11     def predict(self, X):
12         num_test = X.shape[0]
13         Y_pred = np.zeros(num_test, dtype = self.ytr.dtype)
14     
15         for i in range(num_test):
16             distances = np.sum(np.abs(self.Xtr - X[i, :]), axis = 1)
17             min_index = np.argmin(distances)
18             Y_pred[i] = self.ytr[min_index]
19     
20     return Y_pred