generated from LufsX/Challenge-kNN
This commit is contained in:
+3
-3
@@ -39,9 +39,9 @@ name = "陈垣豪"
|
||||
- int: 预测的分类标签。
|
||||
"""
|
||||
|
||||
def classify0(inX, 数据集, 标签, k=3):
|
||||
def classify0(inX, dataSet,label k=3):
|
||||
# 计算协方差矩阵并求其逆矩阵
|
||||
cov_matrix = np.cov(数据集,rowvar=False)
|
||||
cov_matrix = np.cov(dataSet,rowvar=False)
|
||||
inv_cov_matrix = np.linalg.inv(cov_matrix)
|
||||
|
||||
# 计算马氏距离
|
||||
@@ -65,4 +65,4 @@ def classify0(inX, 数据集, 标签, k=3):
|
||||
# 找到出现次数最多的标签
|
||||
predicted_label = max(label_count, key=label_count.get)
|
||||
|
||||
return predicted_label提高准确率
|
||||
return predicted_label
|
||||
|
||||
Reference in New Issue
Block a user