From 50ace15cd5bdd5d784b9fd8791c792d5340d45f7 Mon Sep 17 00:00:00 2001
From: Horo <124332948+threeFeetCat123@users.noreply.github.com>
Date: Fri, 8 Sep 2023 21:00:03 +0800
Subject: [PATCH] Update LinearRegression_scikit-learn.py

changes for python3
---
 LinearRegression/LinearRegression_scikit-learn.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/LinearRegression/LinearRegression_scikit-learn.py b/LinearRegression/LinearRegression_scikit-learn.py
index 8036d58..e82c61a 100644
--- a/LinearRegression/LinearRegression_scikit-learn.py
+++ b/LinearRegression/LinearRegression_scikit-learn.py
@@ -14,7 +14,7 @@ def linearRegression():
     scaler = StandardScaler()   
     scaler.fit(X)
     x_train = scaler.transform(X)
-    x_test = scaler.transform(np.array([1650,3]))
+    x_test = scaler.transform(np.array([[1650,3]]))
     
     # 线性模型拟合
     model = linear_model.LinearRegression()
@@ -39,4 +39,4 @@ def loadnpy_data(fileName):
 
 
 if __name__ == "__main__":
-    linearRegression()
\ No newline at end of file
+    linearRegression()