更新 function.py

function.py

@@ -2,6 +2,7 @@
 import numpy as np
 from collections import Counter
 from scipy.spatial import distance
+from sklearn.preprocessing import StandardScaler
 
 """
 挑战说明:
@@ -87,6 +88,11 @@ if __name__ == "__main__":
     train_data, train_labels = load_data(train_file)
     test_data, test_labels = load_data(test_file)
 
+    # 特征缩放
+    scaler = StandardScaler()
+    train_data = scaler.fit_transform(train_data)
+    test_data = scaler.transform(test_data)
+
     # 测试 classify0 函数
     test_sample = test_data[0]  # 使用测试集中的第一个样本进行测试
     predicted_label = classify0(test_sample, train_data, train_labels)