Skip to content

Instantly share code, notes, and snippets.

@Yuvnish017

Yuvnish017/to_categorical.py

Created July 13, 2021 06:41
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 Yuvnish017/610a57156329b03e2b59cc62a1a54f23 to your computer and use it in GitHub Desktop.
Save Yuvnish017/610a57156329b03e2b59cc62a1a54f23 to your computer and use it in GitHub Desktop.
Download ZIP
OCR
Raw
to_categorical.py
X_train = np.array(X_train)
X_test = np.array(X_test)
Y_train = np.array(Y_train)
Y_test = np.array(Y_test)
Y_train = to_categorical(Y_train)
Y_test = to_categorical(Y_test)
X_train = np.expand_dims(X_train, axis=-1)
X_test = np.expand_dims(X_test, axis=-1)
X_train = X_train/255.
X_test = X_test/255.
print(X_train.shape)
print(X_test.shape)
print(Y_train.shape)
print(Y_test.shape)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment