From cd6439c5347173c34d53ab6b00c04c142bc5874d Mon Sep 17 00:00:00 2001
From: 202210423104 <1961532386@qq.com>
Date: Sun, 3 Nov 2024 16:01:58 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E6=96=87=E4=BB=B6=E8=87=B3?=
 =?UTF-8?q?=20/?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/function.py b/function.py
index 5dbac91..eb0c449 100644
--- a/function.py
+++ b/function.py
@@ -30,14 +30,14 @@ def preprocess_data(X):
     return X_poly
 
 def train_logreg(X, y):
+    X_processed = preprocess_data(X)  # 预处理训练数据
     model = RandomForestClassifier(n_estimators=113, max_depth=10, random_state=42)
-    model.fit(X, y)
+    model.fit(X_processed, y)
     return model  # 返回训练好的模型
 
 def predict_logreg(model, X):
-    return model.predict(X.reshape(1, -1))[0]  # 确保 model 是训练好的模型
+    X_processed = preprocess_data(X.reshape(1, -1))  # 预处理测试数据
+    return model.predict(X_processed)[0]  # 确保 model 是训练好的模型
 
 if __name__ == "__main__":
     X, y = load_data("./data/horse-colic.data")
-    
-    # 这里可能需要确保调用和处理的逻辑符合你的数据处理需求