From 2692f70fc015f40af73ed721aa795ceaa4614135 Mon Sep 17 00:00:00 2001
From: Ulysses-ui <3233559397@qq.com>
Date: Sun, 3 Nov 2024 13:49:21 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

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)