diff --git a/function.py b/function.py index 5aff1fc..51fa106 100644 --- a/function.py +++ b/function.py @@ -65,17 +65,19 @@ def load_data(filepath): - y (np.array): 标签数据。 """ # 使用np.genfromtxt加载数据,并且忽略文件中不能转换为float的行 - data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=np.nan) + with open(filepath, 'r') as file: + lines = file.readlines() - # 删除包含NaN值的行 - data = data[~np.isnan(data).any(axis=1)] - - # 确保数据是二维的 - if data.ndim == 1: - data = data.reshape(1, -1) + 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) X = data[:, :-1] - y = data[:, -1].reshape(-1, 1) # 确保y是二维的 + y = data[:, -1] return X, y