更新 function.py

function.py

@@ -76,7 +76,14 @@ def load_data(filepath):
     - np.array: 特征数据。
     - np.array: 标签数据。
     """
-    data = np.loadtxt(filepath, delimiter=',', dtype=float, skiprows=1)
+    # 使用 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
+
     X = data[:, :-1]  # 所有列，除了最后一列
     y = data[:, -1]   # 最后一列作为标签
     return X, y