Start work on AI
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
import tensorflow as tf
|
||||
import keras
|
||||
print( tf.__version__ )
|
||||
|
||||
from keras.layers import Dense, Flatten, Conv2D
|
||||
from keras import Model
|
||||
|
||||
mnist = keras.datasets.mnist
|
||||
|
||||
(x_train, y_train), (x_test, y_test) = mnist.load_data()
|
||||
x_train, x_test = x_train / 255.0, x_test / 255.0
|
||||
|
||||
# Add a channels dimension
|
||||
x_train = x_train[..., tf.newaxis].astype("float32")
|
||||
x_test = x_test[..., tf.newaxis].astype("float32")
|
||||
|
||||
# batch and shuffle the dataset
|
||||
train_ds = tf.data.Dataset.from_tensor_slices(
|
||||
(x_train, y_train)).shuffle(10000).batch(32)
|
||||
|
||||
test_ds = tf.data.Dataset.from_tensor_slices((x_test, y_test)).batch(32)
|
||||
Reference in New Issue
Block a user