From 9b99bc27482846d11f60ddf89bd270ff00a1770b Mon Sep 17 00:00:00 2001
From: lzh <2233673261@qq.com>
Date: Sat, 2 Nov 2024 20:34:32 +0800
Subject: [PATCH] 444

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

diff --git a/function.py b/function.py
index d2284cc..852db3c 100644
--- a/function.py
+++ b/function.py
@@ -101,12 +101,12 @@ def predict_logreg(inX, weights):
     Returns:
     - int: 预测的分类标签。
     """
-    # 确保输入样本的特征数量与训练时一致
-    if inX.shape[0] != weights.shape[0] - 1:
-        raise ValueError("输入样本的特征数量与训练时不一致")
+    # 确保输入样本是二维数组
+    if inX.ndim == 1:
+        inX = inX.reshape(1, -1)  # 将一维数组转换为二维数组
     
     # 添加偏置项
-    inX = np.insert(inX, 0, 1, axis=1)  # 在特征向量的开头插入偏置项
+    inX = np.hstack([np.ones((inX.shape[0], 1)), inX])  # 在特征向量的开头添加偏置项
     z = np.dot(inX, weights)
     y_pred = 1 / (1 + np.exp(-z))
     return 1 if y_pred > 0.5 else 0