TensorFlow 2快速入门
下载并安装TensorFlow 2.将TensorFlow 2.导入您的程序:
import tensorflow as tf
加载并准备MNIST数据集。将样本从整数转换为浮点数:
mnist = tf.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
x_train.shape
#(60000, 28, 28)
x_test.shape
#(10000, 28, 28)
注:
1、 为什么要/255.0? : 此为归一化操作。
构建模型
tf.keras.Sequential通过堆叠图层来构建模型。选择一个优化器和损失函数进行训练:
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=