diff --git a/function.py b/function.py index ccd0cda..1b8489d 100644 --- a/function.py +++ b/function.py @@ -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])