From 264fbf9da44a894c3e8083fae7a8ba24a20f6a4f Mon Sep 17 00:00:00 2001
From: llll <2992513220@qq.com>
Date: Thu, 31 Oct 2024 23:26:07 +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 | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/function.py b/function.py
index ffa2251..8f71931 100644
--- a/function.py
+++ b/function.py
@@ -44,10 +44,9 @@ def load_data(filepath):
     y = data[:, -1]
     
     # 将非数值数据替换为NaN，并删除包含NaN的行
-    X = np.nan_to_num(X, nan=np.nan)
-    y = np.nan_to_num(y, nan=np.nan)
-    
+    # 注意：这里我们不需要再次使用np.nan_to_num，因为np.genfromtxt已经处理了非数值数据
     # 删除包含NaN的行
+    # np.isnan()函数会返回一个布尔数组，我们用它来创建一个掩码
     mask = ~np.isnan(X).any(axis=1) & ~np.isnan(y)
     X = X[mask]
     y = y[mask]