更新 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
+3 -1
View File
@@ -83,7 +83,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) # 生存
@@ -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]}")