更新 function.py

2024-10-14 21:39:13 +08:00
parent 91c0d80be0
commit aeaa2a4715
1 changed files with 2 additions and 13 deletions
@@ -88,24 +88,13 @@ def load_data(file_path):
    labels = data[:, -1].astype(int)
    return features, labels

-if __name__ == "__main__":
-    train_file = "datingTestSet2.tsv"
-    train_data, train_labels = load_data(train_file)
-    
-    # 预测结果
-    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels, k=3))

-    


 if __name__ == "__main__":
    # 以下代码可用于测试，可自行随意修改
-    train_file = "datingTrainSet2.tsv"
+    train_file = "datingTestSet2.tsv"
    train_data, train_labels = load_data(train_file)
    
-    # 找到最佳 k 值
-    best_k = find_best_k(train_data, train_labels)
-    print(f"Best k: {best_k}")
-    
    # 预测结果
-    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels, best_k))
+    print(classify0(np.array([65915, 3.868359, 0.128078]), train_data, train_labels, k=3))