更新 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
+7 -9
View File
@@ -83,17 +83,15 @@ 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() y.append(int(values[-1]))
# 检查是否存在 '?' 并跳过这些行 X.append([float(x) for x in values[:-1]])
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) return np.array(X), np.array(y)