forked from LufsX/Experiment-LogReg
This commit is contained in:
+10
-8
@@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user