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