diff --git a/function.py b/function.py
index 0c05d08..03d369d 100644
--- a/function.py
+++ b/function.py
@@ -72,9 +72,16 @@ def load_data(filepath):
     - np.array: 特征数据。
     - np.array: 标签数据。
     """
-    data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=0)
+    # 使用numpy的genfromtxt函数加载数据，假设逗号为分隔符
+    data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=np.nan)
+    
+    # 删除包含NaN的行
+    data = data[~np.isnan(data).any(axis=1)]
+    
+    # 将数据转换为numpy数组
     X = data[:, :-1]
     y = data[:, -1].astype(int)  # 将标签列转换为整数类型
+    
     return X, y
 
 if __name__ == "__main__":