上传文件至 /

2024-11-03 16:05:42 +08:00
parent cd6439c534
commit 495a83db7c
1 changed files with 4 additions and 1 deletions
@@ -36,7 +36,10 @@ def train_logreg(X, y):
    return model  # 返回训练好的模型

 def predict_logreg(model, X):
-    X_processed = preprocess_data(X.reshape(1, -1))  # 预处理测试数据
+    # 确保输入数据是一个 numpy 数组
+    if len(X.shape) == 1:
+        X = X.reshape(1, -1)  # 保持输入数据的形状为 (1, n)，符合模型的输入要求
+    X_processed = preprocess_data(X)  # 预处理测试数据
    return model.predict(X_processed)[0]  # 确保 model 是训练好的模型

 if __name__ == "__main__":