Update model.py

gabriellm1 · web-flow · commit c465a06a0d1a · 2021-04-21T17:25:15.000-03:00
diff --git a/src/model.py b/src/model.py
@@ -39,7 +39,8 @@ def train(data, num_estimators, isDataFrame=False):
     X_train, X_test, y_train, y_test = train_test_split(
         X, y, test_size=0.3, random_state=0
     )
-
+    
+    # Regressão Linear
     pipe = Pipeline(
         [
             ("scaler", StandardScaler()),

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,8 @@ def train(data, num_estimators, isDataFrame=False):`
`39`	`39`	`X_train, X_test, y_train, y_test = train_test_split(`
`40`	`40`	`X, y, test_size=0.3, random_state=0`
`41`	`41`	`)`
`42`		`-`
	`42`	`+`
	`43`	`+ # Regressão Linear`
`43`	`44`	`pipe = Pipeline(`
`44`	`45`	`[`
`45`	`46`	`("scaler", StandardScaler()),`