NearestNeighbor

最新推荐文章于 2025-03-20 09:57:37 发布

Binge_van

最新推荐文章于 2025-03-20 09:57:37 发布

阅读量351

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/Binge_van/article/details/85230339

本文介绍了一个基于Python的近邻算法实现，通过定义NearestNeighbor类，该类包含训练和预测方法，实现对输入数据的分类预测。文章详细展示了算法的工作原理，包括距离计算和类别预测过程，以及算法的时间复杂度分析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np

class NearestNeighbor:

  	def __init__(self):
		pass
	
	def train(self, X, y):
		self.Xtr = X
		self.ytr = y
	
	def predict(self, X):
		num_test = X.shape[0]
		Ypred = np.zeros(num_tset, dtype=self.ytr.dtype)
	
	for i in range(num_test):
		distance = np.sum(np.abs(self.Xtr - [i,:]), axis=1)
		min_index = np.argmin(distance)
		Ypred[i] = self.ytr[min_index]

	return Ypred