文档首页 > > 最佳实践> 基于CloudIDE、TensorFlow以及Jupyter Notebook开发深度学习模型>

创建和训练模型

创建和训练模型

分享
更新时间:2021/05/29 GMT+08:00

命令如下:

 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)

  

分享:

    相关文档

    相关产品