写一段卷积神经网络的代码
时间: 2023-10-15 14:06:00 浏览: 138
抱歉,作为AI语言模型,我无法编写代码。但是,以下是一个简单的卷积神经网络示例:
```python
import tensorflow as tf
# 定义输入层
inputs = tf.keras.layers.Input(shape=(28, 28, 1))
# 第一层卷积
x = tf.keras.layers.Conv2D(32, (3, 3), activation='relu')(inputs)
x = tf.keras.layers.MaxPooling2D()(x)
# 第二层卷积
x = tf.keras.layers.Conv2D(64, (3, 3), activation='relu')(x)
x = tf.keras.layers.MaxPooling2D()(x)
# 第三层卷积
x = tf.keras.layers.Conv2D(128, (3, 3), activation='relu')(x)
# 展平层
x = tf.keras.layers.Flatten()(x)
# 全连接层
x = tf.keras.layers.Dense(128, activation='relu')(x)
# 输出层
outputs = tf.keras.layers.Dense(10, activation='softmax')(x)
# 定义模型
model = tf.keras.Model(inputs=inputs, outputs=outputs)
# 编译模型
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
# 训练模型
model.fit(train_images, train_labels, epochs=10, validation_data=(test_images, test_labels))
```
阅读全文