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 分类算法,使用马氏距离。
|
KNN 分类算法,使用马氏距离。
|
||||||
|
|
||||||
@@ -54,7 +54,6 @@ def classify0(inX, dataSet, labels, k=100, cov_inv=None):
|
|||||||
- int: 预测的分类标签。
|
- int: 预测的分类标签。
|
||||||
"""
|
"""
|
||||||
if cov_inv is None:
|
if cov_inv is None:
|
||||||
# 计算训练数据的协方差矩阵并求逆
|
|
||||||
cov_matrix = np.cov(dataSet.T)
|
cov_matrix = np.cov(dataSet.T)
|
||||||
cov_inv = inv(cov_matrix)
|
cov_inv = inv(cov_matrix)
|
||||||
differences = dataSet - inX
|
differences = dataSet - inX
|
||||||
|
|||||||
Reference in New Issue
Block a user