generated from LufsX/Challenge-kNN
This commit is contained in:
+1
-2
@@ -39,7 +39,7 @@ name = "吴俊煯"
|
||||
|
||||
|
||||
# 这里写自己的代码实现
|
||||
def classify0(inX, dataSet, labels, k=100, cov_inv=None):
|
||||
def classify0(inX, dataSet, labels, k=60, cov_inv=None):
|
||||
"""
|
||||
KNN 分类算法,使用马氏距离。
|
||||
|
||||
@@ -54,7 +54,6 @@ def classify0(inX, dataSet, labels, k=100, cov_inv=None):
|
||||
- int: 预测的分类标签。
|
||||
"""
|
||||
if cov_inv is None:
|
||||
# 计算训练数据的协方差矩阵并求逆
|
||||
cov_matrix = np.cov(dataSet.T)
|
||||
cov_inv = inv(cov_matrix)
|
||||
differences = dataSet - inX
|
||||
|
||||
Reference in New Issue
Block a user