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()]
|
parts = [part.strip() for part in line.split()]
|
||||||
# 将缺失值'?'替换为0(或者你可以选择其他适当的处理方式)
|
# 将缺失值'?'替换为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的话)
|
# 将标签从2转换为1(如果原始标签是1和2的话)
|
||||||
label = int(parts[-1]) - 1 # 假设原始标签是1和2,转换为0和1
|
label = int(parts[-1]) - 1 # 假设原始标签是1和2,转换为0和1
|
||||||
data.append(features + [label])
|
data.append(features + [label])
|
||||||
|
|||||||
Reference in New Issue
Block a user