From 6f8e5cc23a3f16366f2dc0fe89c163df89684bf2 Mon Sep 17 00:00:00 2001
From: leiwu <1929652729@qq.com>
Date: Sun, 3 Nov 2024 20:02:28 +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 | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/function.py b/function.py
index 407f332..44db30f 100644
--- a/function.py
+++ b/function.py
@@ -82,8 +82,10 @@ def load_data(filepath):
     - np.array: 特征数据。
     - np.array: 标签数据。
     """
-    # 使用 genfromtxt 来加载数据，并处理缺失值和错误的数据格式
+    # 使用 genfromtxt 来加载数据，忽略无法转换的值，并处理缺失值
     data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=np.nan)
+    # 移除含有NaN的行
+    data = data[~np.isnan(data).any(axis=1)]
     X = data[:, :-1]
     y = data[:, -1].astype(int)
     return X, y