Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- from scipy.spatial.distance import cdist
- def nearest_neighbor(samples, targets, samples_to_classify, metric='euclidean'):
- return targets[np.argmin(cdist(samples_to_classify, samples, metric=metric), axis=1)]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement