From f89d0a6292dcc78a4c64cd7e8ee590f6bf23590d Mon Sep 17 00:00:00 2001
From: heart20003 <1789588961@qq.com>
Date: Sun, 3 Nov 2024 20:58:28 +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 | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/function.py b/function.py
index ba8d467..cf46fa8 100644
--- a/function.py
+++ b/function.py
@@ -35,16 +35,13 @@ def load_data(filepath):
     y = data[:, -1]
     return X, y
 
-def train_logreg(X, y, initial_learning_rate=0.1, iterations=1000, decay_rate=0.01):
-    """训练逻辑回归模型，使用随机梯度下降和学习率衰减."""
+def train_logreg(X, y, learning_rate=0.1, iterations=1000):
+    """训练逻辑回归模型，使用随机梯度下降."""
     n_samples, n_features = X.shape
     weights = np.zeros(n_features + 1)  # 增加一位用于偏置
     X = np.c_[X, np.ones((n_samples, 1))]  # 增加偏置项
 
-    for i in range(iterations):
-        # 更新学习率
-        learning_rate = initial_learning_rate / (1 + decay_rate * i)
-
+    for _ in range(iterations):
         random_index = np.random.randint(n_samples)
         X_i = X[random_index]
         y_i = y[random_index]