# Fit the scaler on the training data
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()
scaler.fit(df_train)
features = df_train.columns

# Scale the training, test, and validation sets
features = df_train.columns

df_train[features] = scaler.transform(df_train[features])
df_val[features] = scaler.transform(df_val[features])
df_test[features] = scaler.transform(df_test[features])

df_train.head()