diff --git a/function.py b/function.py index 9f58aef..1e8ce8e 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) # 生存 @@ -104,6 +104,7 @@ def load_data(filepath): return data, labels + def preprocess_data(X): """特征标准化。""" scaler = StandardScaler()