更新 function.py

function.py

@@ -40,13 +40,14 @@ def load_data(filepath):
     data = np.genfromtxt(filepath, delimiter=',', skip_header=1, filling_values=np.nan)
     
     # 分离特征和标签
+    # 确保data不是一维数组
+    if data.ndim == 1:
+        data = data.reshape(1, -1)  # 将其转换为二维数组，如果它是一维的
+
     X = data[:, :-1]
-    y = data[:, -1]
-    
-    # 将非数值数据替换为NaN，并删除包含NaN的行
-    # 注意：这里我们不需要再次使用np.nan_to_num，因为np.genfromtxt已经处理了非数值数据
+    y = data[:, -1].flatten()  # 确保y是一维数组
+
     # 删除包含NaN的行
-    # np.isnan()函数会返回一个布尔数组，我们用它来创建一个掩码
     mask = ~np.isnan(X).any(axis=1) & ~np.isnan(y)
     X = X[mask]
     y = y[mask]