更新 function.py

2024-10-14 21:18:57 +08:00
parent 458d75187d
commit 84bab8003a
1 changed files with 10 additions and 2 deletions
@@ -1,5 +1,6 @@
 # function.py
 import numpy as np
+from sklearn.preprocessing import StandardScaler

 """
 挑战说明:
@@ -54,7 +55,12 @@ def classify0(inX, dataSet, labels, k=3):
    Returns:
    - int: 预测的分类标签。
    """
-# 计算协方差矩阵的逆
+    # 标准化数据
+    scaler = StandardScaler()
+    dataSet = scaler.fit_transform(dataSet)
+    inX = scaler.transform([inX])[0]
+    
+    # 计算协方差矩阵的逆
    cov_matrix = np.cov(dataSet, rowvar=False)
    inv_cov_matrix = np.linalg.inv(cov_matrix)
    
@@ -84,4 +90,6 @@ if __name__ == "__main__":
    # 以下代码可用于测试，可自行随意修改
    train_file = "datingTrainSet2.tsv"
    train_data, train_labels = load_data(train_file)
-    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels))
+
+    # 预测结果
+    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels, k=3))