更新 function.py

2024-11-06 14:57:49 +08:00
parent bc7e70a92a
commit 8d08341ff9
1 changed files with 5 additions and 2 deletions
@@ -41,8 +41,11 @@ def load_data(train_path, test_path):
    Returns:
        tuple: 包含训练数据特征、训练数据标签、测试数据特征、测试数据标签
    """
-    train_data = pd.read_csv("./data/horse-colic.data", header=None, na_values='?')
-    test_data = pd.read_csv("./data/horse-colic.test", header=None, na_values='?')
+    train_path = "./data/horse-colic.data"
+    test_path = "./data/horse-colic.test"
+    train_X, train_y, test_X, test_y = load_data(train_path, test_path)
+    train_data = pd.read_csv(train_path, header=None, na_values='?')
+    test_data = pd.read_csv(test_path, header=None, na_values='?')

    # 处理缺失值，这里简单删除含有缺失值的行
    train_data = train_data.dropna()