diff --git a/function.py b/function.py index 55a5034..5764230 100644 --- a/function.py +++ b/function.py @@ -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 分类算法,使用马氏距离。 @@ -53,8 +53,7 @@ def classify0(inX, dataSet, labels, k=100, cov_inv=None): Returns: - int: 预测的分类标签。 """ - if cov_inv is None: - # 计算训练数据的协方差矩阵并求逆 + if cov_inv is None: cov_matrix = np.cov(dataSet.T) cov_inv = inv(cov_matrix) differences = dataSet - inX