main
Gasper Spagnolo 2022-09-05 20:37:53 +02:00
parent 01f2fc36b9
commit fa85c8f265
1 changed files with 2 additions and 2 deletions

View File

@ -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