更新 function.py

function.py

@@ -36,11 +36,24 @@ def load_data(filepath):
     - np.array: 特征数据。
     - np.array: 标签数据。
     """
-    data = np.loadtxt(filepath, delimiter=',', skiprows=1)
+    # 使用np.genfromtxt加载数据，它可以自动处理缺失值
+    data = np.genfromtxt(filepath, delimiter=',', skip_header=1, filling_values=np.nan)
+    
+    # 分离特征和标签
     X = data[:, :-1]
     y = data[:, -1]
+    
+    # 将非数值数据替换为NaN，并删除包含NaN的行
+    X = np.nan_to_num(X, nan=np.nan)
+    y = np.nan_to_num(y, nan=np.nan)
+    
+    # 删除包含NaN的行
+    mask = ~np.isnan(X).any(axis=1) & ~np.isnan(y)
+    X = X[mask]
+    y = y[mask]
+    
     return X, y
-
+    
 def sigmoid(z):
     """
     Sigmoid函数。