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

This commit is contained in:
2024-11-03 16:27:03 +08:00
parent d9c30f0b4c
commit fe33d9accb
+9 -7
View File
@@ -83,15 +83,17 @@ def load_data(filepath):
- np.array: 特征数据。
- np.array: 标签数据。
"""
with open(filepath, 'r') as f:
lines = f.readlines()
X = []
y = []
for line in lines:
values = line.strip().split()
y.append(int(values[-1]))
X.append([float(x) for x in values[:-1]])
with open(filepath, 'r') as f:
for line in f:
values = line.strip().split()
# 检查是否存在 '?' 并跳过这些行
if '?' in values:
continue
# 将标签和特征转换为浮点数
y.append(int(values[-1]))
X.append([float(x) for x in values[:-1]])
return np.array(X), np.array(y)