更新 function.py

function.py

@@ -76,16 +76,22 @@ def load_data(filepath):
     - np.array: 特征数据。
     - np.array: 标签数据。
     """
-    # 使用 np.genfromtxt 来处理缺失值
     data = np.genfromtxt(filepath, delimiter=',', dtype=float, skip_header=1, 
                          filling_values=np.nan,  # 用 NaN 填充缺失值
                          missing_values='?')     # 将问号识别为缺失值
 
-    # 处理缺失值（例如将 NaN 替换为 0 或者其他值）
+    # 检查数据的形状
-    data = np.nan_to_num(data, nan=0.0)  # 将 NaN 替换为 0.0
+    print("Data shape:", data.shape)
+
+    # 检查是否为一维数组
+    if data.ndim == 1:
+        # 如果是一维数组，假设最后一列为标签，前面的部分作为特征
+        X = data[:-1].reshape(-1, 1)  # 变为二维数组，确保有一个特征维度
+        y = data[-1]  # 只有一个标签
+    else:
+        X = data[:, :-1]  # 所有列，除了最后一列
+        y = data[:, -1]   # 最后一列作为标签
 
-    X = data[:, :-1]  # 所有列，除了最后一列
-    y = data[:, -1]   # 最后一列作为标签
     return X, y
 
 if __name__ == "__main__":