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

最新推荐文章于 2024-12-30 21:21:59 发布

原创

最新推荐文章于 2024-12-30 21:21:59 发布 · 1.9k 阅读

标签

#深度学习 #tensorflow

收录于

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

逐层增加

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')