diff --git a/function.py b/function.py index 51fa106..4094fb3 100644 --- a/function.py +++ b/function.py @@ -64,18 +64,10 @@ def load_data(filepath): - X (np.array): 特征数据。 - y (np.array): 标签数据。 """ - # 使用np.genfromtxt加载数据,并且忽略文件中不能转换为float的行 - with open(filepath, 'r') as file: - lines = file.readlines() - - data = [] - for line in lines: - split_line = line.strip().split() - # 将非数值数据转换为0 - split_line = [float(x) if x else 0 for x in split_line] - data.append(split_line) - - data = np.array(data) + data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=0) + # 将问号 '?' 替换为 NaN,并将 NaN 替换为 0 + data = np.where(data == '?', np.nan, data) + data = np.where(np.isnan(data), 0, data) X = data[:, :-1] y = data[:, -1] return X, y