generated from LufsX/Experiment-LogReg
This commit is contained in:
+5
-3
@@ -81,15 +81,15 @@ def load_data(filepath):
|
||||
values = line.strip().replace('?', '0').split()
|
||||
try:
|
||||
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) # 生存
|
||||
else:
|
||||
continue # 无效标签,丢弃
|
||||
|
||||
|
||||
# 用0填充缺失值
|
||||
values = [0 if v == 0 else v for v in values] # 直接用0填充
|
||||
data.append(values[:2] + values[3:23]) # 保留特征1, 2 和特征4-23
|
||||
@@ -101,6 +101,7 @@ def load_data(filepath):
|
||||
|
||||
return data, labels
|
||||
|
||||
|
||||
def preprocess_data(X):
|
||||
"""特征标准化。"""
|
||||
scaler = StandardScaler()
|
||||
@@ -123,3 +124,4 @@ if __name__ == "__main__":
|
||||
# 测试单个数据
|
||||
y_pred = predict_logreg(X_test[11], weights)
|
||||
print(f"预测结果: {y_pred}, 实际结果: {y_test[11]}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user