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.
|
# Standardize features by removing the mean and scaling to unit variance.
|
||||||
scale = StandardScaler()
|
scale = StandardScaler()
|
||||||
x_train = scale.fit_transform(x_train)
|
X_train = scale.fit_transform(X_train)
|
||||||
x_test = scale.transform(x_test)
|
X_test = scale.transform(X_test)
|
||||||
|
|
||||||
return X_train, X_test, y_train, y_test
|
return X_train, X_test, y_train, y_test
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue