generated from LufsX/Experiment-LogReg
This commit is contained in:
+2
-1
@@ -86,7 +86,7 @@ def load_data(filepath):
|
|||||||
values = [float(v) for v in values]
|
values = [float(v) for v in values]
|
||||||
|
|
||||||
# 合并特征23的取值
|
# 合并特征23的取值
|
||||||
if values[22] == 2 or values[22] == 3: # 特征23(索引22)
|
if values[22] == 2 or values[22] == 3: # 特征23(索引22)
|
||||||
labels.append(0) # 死亡
|
labels.append(0) # 死亡
|
||||||
elif values[22] == 1:
|
elif values[22] == 1:
|
||||||
labels.append(1) # 生存
|
labels.append(1) # 生存
|
||||||
@@ -104,6 +104,7 @@ def load_data(filepath):
|
|||||||
|
|
||||||
return data, labels
|
return data, labels
|
||||||
|
|
||||||
|
|
||||||
def preprocess_data(X):
|
def preprocess_data(X):
|
||||||
"""特征标准化。"""
|
"""特征标准化。"""
|
||||||
scaler = StandardScaler()
|
scaler = StandardScaler()
|
||||||
|
|||||||
Reference in New Issue
Block a user