From 820027e8945e30f3b4480709b2cc5be5b18e3e67 Mon Sep 17 00:00:00 2001 From: DREMN2112 Date: Thu, 7 Nov 2024 16:47:15 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- function.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/function.py b/function.py index 262c0cf..9720f59 100644 --- a/function.py +++ b/function.py @@ -69,7 +69,7 @@ def load_data(filepath): """ data = [] labels = [] - + with open(filepath, 'r') as file: for line in file: line = line.strip().split() @@ -85,15 +85,20 @@ def load_data(filepath): continue # 将特征转换为浮点数 - features = list(map(float, line[:-1])) - label = int(line[-1]) + try: + features = list(map(float, line[:-1])) # 所有特征转换为浮点数 + label = int(line[-1]) # 标签为整数 + except ValueError: + # 如果遇到无法转换的值，则跳过这一行 + continue # 添加到数据集 data.append(features) labels.append(label) - + return np.array(data), np.array(labels) + if __name__ == "__main__": # 加载训练集 X_train, y_train = load_data("./data/horse-colic.data")