更新 function.py
Gitea Actions Experiment Test / Experiment-Actions (push) Successful in 12s

This commit is contained in:
2024-11-03 20:02:28 +08:00
parent 6337178f7c
commit 6f8e5cc23a
+3 -1
View File
@@ -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