Skip to content

Instantly share code, notes, and snippets.

@jeetaf
Created July 13, 2021 17:12
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jeetaf/98104baa27d2d1f21e977c0634dd3cc4 to your computer and use it in GitHub Desktop.
Save jeetaf/98104baa27d2d1f21e977c0634dd3cc4 to your computer and use it in GitHub Desktop.
X_train, X_test, y_train, y_test = train_test_split(x_ros,y_ros,test_size=0.2,random_state=42)
y_train=to_categorical(y_train)
y_test=to_categorical(y_test)
print('X_train :',X_train.shape)
print('y_train :',y_train.shape)
print('X_test :',X_test.shape)
print('y_test :',y_test.shape)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment