Skip to content

Instantly share code, notes, and snippets.

@douglaspsteen
Created September 5, 2020 18:35
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save douglaspsteen/a03f7ec1a9f84d31e0bf56c7afa0fa53 to your computer and use it in GitHub Desktop.
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
# Split the data - 75% train, 25% test
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25,
random_state=1)
# Scale the X data
scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment