diff --git a/function.py b/function.py
index 69b1916..103d776 100644
--- a/function.py
+++ b/function.py
@@ -84,10 +84,17 @@ def load_data(filepath):
     with open(filepath, 'r') as file:
         for line in file:
             fields = line.strip().split()
-            if '?' in fields:
-                continue  # 跳过包含缺失值的行
-            data.append([float(x) for x in fields[:-1]])
-            labels.append(int(fields[-1]))  # 假设标签在最后一列
+            # 检查是否有缺失值，并排除不符合行长度要求的数据
+            if '?' in fields or len(fields) < 2:
+                continue
+            try:
+                # 假设特征在前面，标签在最后一列
+                data.append([float(x) for x in fields[:-1]])
+                labels.append(int(fields[-1]))
+            except ValueError:
+                # 如果无法转换为浮点数或整数，跳过该行
+                continue
+    # 返回值需要是两个NumPy数组
     return np.array(data), np.array(labels)