diff --git a/function.py b/function.py
index 567963e..5aff1fc 100644
--- a/function.py
+++ b/function.py
@@ -64,9 +64,18 @@ def load_data(filepath):
     - X (np.array): 特征数据。
     - y (np.array): 标签数据。
     """
-    data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=0)
+    # 使用np.genfromtxt加载数据，并且忽略文件中不能转换为float的行
+    data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=np.nan)
+    
+    # 删除包含NaN值的行
+    data = data[~np.isnan(data).any(axis=1)]
+    
+    # 确保数据是二维的
+    if data.ndim == 1:
+        data = data.reshape(1, -1)
+    
     X = data[:, :-1]
-    y = data[:, -1]
+    y = data[:, -1].reshape(-1, 1)  # 确保y是二维的
     return X, y