更新 function.py
Gitea Actions Challenge Test / Challenge-Actions (push) Successful in 12s

This commit is contained in:
2024-10-11 11:14:44 +08:00
parent d197c48d00
commit 798afa851f
-6
View File
@@ -1,7 +1,5 @@
# function.py # function.py
pip install sklearn
import numpy as np import numpy as np
from sklearn.preprocessing import StandardScaler
from numpy.linalg import inv from numpy.linalg import inv
""" """
挑战说明: 挑战说明:
@@ -59,10 +57,6 @@ def classify0(inX, dataSet, labels, k=4, cov_inv=None):
# 计算训练数据的协方差矩阵并求逆 # 计算训练数据的协方差矩阵并求逆
cov_matrix = np.cov(dataSet.T) cov_matrix = np.cov(dataSet.T)
cov_inv = inv(cov_matrix) cov_inv = inv(cov_matrix)
# 计算待分类样本与所有训练样本之间的马氏距离
# 注意:这里我们假设 inX 已经被适当地预处理(例如,通过 StandardScaler 归一化)
# 并且 dataSet 也是用相同的预处理步骤处理的
differences = dataSet - inX differences = dataSet - inX
mahalanobis_distances = np.sqrt(np.sum(differences @ cov_inv @ differences.T, axis=1)) mahalanobis_distances = np.sqrt(np.sum(differences @ cov_inv @ differences.T, axis=1))