From 09ebfabb61c804e826d756279a2b048d91210e62 Mon Sep 17 00:00:00 2001
From: hao <19382120447@163.com>
Date: Thu, 31 Oct 2024 17:13:24 +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 | 16 +++-------------
 1 file changed, 3 insertions(+), 13 deletions(-)

diff --git a/function.py b/function.py
index a03cb8c..0c05d08 100644
--- a/function.py
+++ b/function.py
@@ -1,5 +1,4 @@
 import numpy as np
-import pandas as pd
 
 """
 实验说明:
@@ -73,18 +72,9 @@ def load_data(filepath):
     - np.array: 特征数据。
     - np.array: 标签数据。
     """
-    # 使用pandas读取数据，假设逗号为分隔符
-    data = pd.read_csv(filepath, header=None)
-    
-    # 将所有非数值列转换为数值型，非数值的转换为NaN，然后填充NaN为0
-    for col in data.columns:
-        if data[col].dtype == 'object':
-            data[col] = pd.to_numeric(data[col], errors='coerce').fillna(0)
-    
-    # 将数据转换为numpy数组
-    X = data.iloc[:, :-1].values
-    y = data.iloc[:, -1].values
-    
+    data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=0)
+    X = data[:, :-1]
+    y = data[:, -1].astype(int)  # 将标签列转换为整数类型
     return X, y
 
 if __name__ == "__main__":