diff --git a/function.py b/function.py index 4b3bde6..06e1b31 100644 --- a/function.py +++ b/function.py @@ -102,7 +102,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 labels.append(0) # 死亡 elif values[22] == 1: labels.append(1) # 生存