深度学习 - 建立模型的三种方式

壮壮不太胖^QwQ

于 2020-09-14 22:52:00 发布

阅读量1.8k

点赞数 1

文章标签：深度学习 tensorflow

本文链接：https://blog.csdn.net/weixin_46072771/article/details/108589766

版权

本文介绍了深度学习中建立模型的三种方式：逐层增加、一次构建完成和函数API形式。函数API形式允许创建多输入或输出的模型，如在图像定位任务中用于锁狗头的深度学习模型。

摘要由CSDN通过智能技术生成

逐层增加

model = tf.keras.Sequential()  

model.add(tf.keras.layers.Flatten(input_shape=(28, 28)))
model.add(tf.keras.layers.Dense(32, activation='relu'))
model.add(tf.keras.layers.Dropout(0.5))
model.add(tf.keras.layers.Dense(64, activation='relu'))

model.add(keras.layers.Dense(10, activation='softmax'))