更新 function.py

function.py

@@ -93,22 +93,15 @@ def load_data(filepath):
     - X (np.array): 特征数据。
     - y (np.array): 标签数据。
     """
-    # 检查文件是否存在
-    if not os.path.exists(filepath):
-        raise FileNotFoundError(f"文件 {filepath} 不存在")
-
-    # 使用pandas读取数据，将问号替换为NaN
-    data = pd.read_csv(filepath, header=None, na_values='?')
+    # 使用numpy的genfromtxt函数读取数据，将问号替换为NaN
+    data = np.genfromtxt(filepath, delimiter=',', invalid_raise=False, filling_values=np.nan)
     
-    # 将所有列转换为数值类型，非数值列将被设置为NaN
-    data = data.apply(pd.to_numeric, errors='coerce')
-
     # 丢弃完全为NaN的行
-    data = data.dropna()
+    data = data[~np.isnan(data).any(axis=1)]
 
     # 分离特征和标签
-    X = data.iloc[:, :-1].values
-    y = data.iloc[:, -1].values
+    X = data[:, :-1]
+    y = data[:, -1]
 
     return X, y