Start work on AI

This commit is contained in:
2026-02-09 20:15:21 -08:00
parent f77c2a085a
commit cc0ac9ffa9
+21
View File
@@ -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)