diff --git a/function.py b/function.py
index 2e3270a..a0d5613 100644
--- a/function.py
+++ b/function.py
@@ -50,7 +50,7 @@ if __name__ == "__main__":
     model = train_logreg(X_train, y_train)
 
     # 预测
-    y_pred = model.predict(preprocess_data(X_test))  # 直接在测试集上预测
+    y_pred = predict_logreg(model, X_test)  # 调用预测函数
 
     # 计算准确率
     accuracy = np.mean(y_pred == y_test)