diff --git a/function.py b/function.py index 8ed6546..5b19647 100644 --- a/function.py +++ b/function.py @@ -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