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

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