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

This commit is contained in:
Wyy
2024-11-02 20:42:19 +08:00
parent 3a40f5d6b2
commit e062dd40bd
+10 -8
View File
@@ -65,17 +65,19 @@ def load_data(filepath):
- y (np.array): 标签数据。 - y (np.array): 标签数据。
""" """
# 使用np.genfromtxt加载数据,并且忽略文件中不能转换为float的行 # 使用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 = data[~np.isnan(data).any(axis=1)] for line in lines:
split_line = line.strip().split()
# 确保数据是二维的 # 将非数值数据转换为0
if data.ndim == 1: split_line = [float(x) if x else 0 for x in split_line]
data = data.reshape(1, -1) data.append(split_line)
data = np.array(data)
X = data[:, :-1] X = data[:, :-1]
y = data[:, -1].reshape(-1, 1) # 确保y是二维的 y = data[:, -1]
return X, y return X, y