diff --git a/function.py b/function.py index 86186f8..3b47fca 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) # 生存 @@ -143,4 +143,4 @@ if __name__ == "__main__": # 测试单个数据 y_pred = predict_logreg(X_test[11], weights) - print(f"预测结果: {y_pred}, 实际结果: {y_test[11]}") + print(f"预测结果: {y_pred}, 实际结果: {y_test[11]}") \ No newline at end of file