更新 function.py

2024-11-03 16:07:37 +08:00
parent 495a83db7c
commit 2efb7601a1
1 changed files with 1 additions and 1 deletions
@@ -35,7 +35,7 @@ def train_logreg(X, y):
    model.fit(X_processed, y)
    return model  # 返回训练好的模型

-def predict_logreg(model, X):
+def predict_logreg(X, model):
    # 确保输入数据是一个 numpy 数组
    if len(X.shape) == 1:
        X = X.reshape(1, -1)  # 保持输入数据的形状为 (1, n)，符合模型的输入要求