diff --git a/function.py b/function.py index abb219c..86186f8 100644 --- a/function.py +++ b/function.py @@ -86,7 +86,7 @@ def load_data(filepath): values = [float(v) for v in values] # 合并特征23的取值 - if values[22] == 2 or values[22] == 3: # 特征23(索引22) + if values[22] == 2 or values[22] == 3: # 特征23(索引22) labels.append(0) # 死亡 elif values[22] == 1: labels.append(1) # 生存