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

This commit is contained in:
2024-11-07 16:47:15 +08:00
parent cc4498733b
commit 820027e894
+7 -2
View File
@@ -85,8 +85,12 @@ def load_data(filepath):
continue continue
# 将特征转换为浮点数 # 将特征转换为浮点数
features = list(map(float, line[:-1])) try:
label = int(line[-1]) features = list(map(float, line[:-1])) # 所有特征转换为浮点数
label = int(line[-1]) # 标签为整数
except ValueError:
# 如果遇到无法转换的值,则跳过这一行
continue
# 添加到数据集 # 添加到数据集
data.append(features) data.append(features)
@@ -94,6 +98,7 @@ def load_data(filepath):
return np.array(data), np.array(labels) return np.array(data), np.array(labels)
if __name__ == "__main__": if __name__ == "__main__":
# 加载训练集 # 加载训练集
X_train, y_train = load_data("./data/horse-colic.data") X_train, y_train = load_data("./data/horse-colic.data")