更新 function.py

2024-11-02 21:28:38 +08:00
parent 528ab44a98
commit 1c0dd4f373
1 changed files with 3 additions and 2 deletions
@@ -92,8 +92,9 @@ def load_data(filepath):
 # main函数，程序的入口点
 def main():
-    X_train, y_train = load_data("./data/horse-colic.data")  # 加载训练数据
+    # 假设data文件夹和function.py在同一目录下
-    X_test, y_test = load_data("./data/horse-colic.test")  # 加载测试数据
+    X_train, y_train = load_data("./data/horse-colic.data")
    X_test, y_test = load_data("./data/horse-colic.test")
    if X_train is not None and X_test is not None:
        # 特征缩放