Updatre
parent
01f2fc36b9
commit
fa85c8f265
4
main.py
4
main.py
|
@ -118,8 +118,8 @@ def split_data(df, split_value):
|
|||
|
||||
# Standardize features by removing the mean and scaling to unit variance.
|
||||
scale = StandardScaler()
|
||||
x_train = scale.fit_transform(x_train)
|
||||
x_test = scale.transform(x_test)
|
||||
X_train = scale.fit_transform(X_train)
|
||||
X_test = scale.transform(X_test)
|
||||
|
||||
return X_train, X_test, y_train, y_test
|
||||
|
||||
|
|
Loading…
Reference in New Issue