From fba273d85b6c2e6c3418c9b4873eab99cc5136f8 Mon Sep 17 00:00:00 2001
From: Frae19 <515745323@qq.com>
Date: Tue, 5 Nov 2024 15:10:04 +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 | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/function.py b/function.py
index bb45560..73f3559 100644
--- a/function.py
+++ b/function.py
@@ -91,22 +91,23 @@ def load_data(filepath):
     - directory (str): 数据文件路径。
 
     Returns:
-    - Tuple[np.array, np.array]: 特征数据和标签数据。
+    - np.array: 特征数据。
+    - np.array: 标签数据。
     """
     data = []
     with open(filepath, 'r') as f:
         for line in f:
             row = []
             values = line.strip().split()
-            if '?' in values:
-                continue  # Ignore lines with missing data
             for value in values:
-                row.append(float(value))
+                try:
+                    row.append(float(value))
+                except ValueError:
+                    row.append(-1)
             data.append(row)
     np_data = np.array(data)
     X = np_data[:, :-1]
     y = np_data[:, -1]
-    y = np.array([0 if label == 1 else (1 if label == 2 else -1) for label in y])
     return X, y
 
 if __name__ == "__main__":