From d20f22777f143ad416617cc815d793e0cc79de30 Mon Sep 17 00:00:00 2001
From: 202210423104 <1961532386@qq.com>
Date: Sun, 3 Nov 2024 16:25:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/function.py b/function.py
index 13f23fe..2e3270a 100644
--- a/function.py
+++ b/function.py
@@ -37,8 +37,8 @@ def train_logreg(X, y):
     return model  # 返回训练好的模型
 
 def predict_logreg(model, X):
-    X_processed = preprocess_data(X.reshape(1, -1))  # 预处理单个测试样本
-    return model.predict(X_processed)[0]  # 返回预测结果
+    X_processed = preprocess_data(X)  # 预处理输入数据
+    return model.predict(X_processed)  # 返回预测结果
 
 if __name__ == "__main__":
     X, y = load_data("./data/horse-colic.data")
@@ -50,9 +50,7 @@ if __name__ == "__main__":
     model = train_logreg(X_train, y_train)
 
     # 预测
-    y_pred = []
-    for sample in X_test:
-        y_pred.append(predict_logreg(model, sample))  # 逐个样本预测
+    y_pred = model.predict(preprocess_data(X_test))  # 直接在测试集上预测
 
     # 计算准确率
     accuracy = np.mean(y_pred == y_test)