generated from LufsX/Experiment-LogReg
This commit is contained in:
+1
-1
@@ -60,7 +60,7 @@ def load_data(filepath):
|
||||
# 分割行,并去除任何可能的空白字符
|
||||
parts = [part.strip() for part in line.split()]
|
||||
# 将缺失值'?'替换为0(或者你可以选择其他适当的处理方式)
|
||||
features = [float(part) if part != '?' else 0 for part in parts[:-1]]
|
||||
features = [float(part) if part != '?' else 1 for part in parts[:-1]]
|
||||
# 将标签从2转换为1(如果原始标签是1和2的话)
|
||||
label = int(parts[-1]) - 1 # 假设原始标签是1和2,转换为0和1
|
||||
data.append(features + [label])
|
||||
|
||||
Reference in New Issue
Block a user