更新 function.py

2024-11-03 20:52:49 +08:00
parent 95639a294d
commit b21fdffd5b
1 changed files with 1 additions and 0 deletions
@@ -37,6 +37,7 @@ def train_logreg(X, y, learning_rate=0.001, iterations=10000, batch_size=64, lam
            
        if i % 1000 == 0:  # 每1000次迭代打印一次信息
            print(f"迭代 {i}: 当前权重形状: {weights.shape}")
+
    return weights

 def predict_logreg(inX, weights):