更新 function.py
Gitea Actions Experiment Test / Experiment-Actions (push) Successful in 25s

This commit is contained in:
2024-11-02 14:13:04 +08:00
parent 61367c616d
commit e2f5b6ad04
+5 -3
View File
@@ -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]}")