CodeArts IDE Online
CodeArts IDE Online
本文导读
展开导读
链接复制成功!
创建和训练模型
命令如下:
1 2 3 4 5 6 7 8 9 10 11 |
# create model model = keras.Sequential([ keras.layers.Flatten(input_shape=(28, 28)), keras.layers.Dense(128, activation='relu'), keras.layers.Dense(10) ]) model.compile(optimizer='adam', loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True), metrics=['accuracy']) # training model.fit(train_images, train_labels, epochs=10) |