帮我写一个卷积神经网络代码
时间: 2023-10-23 18:09:59 浏览: 40
好的,以下是一个简单的卷积神经网络代码:
```python
import tensorflow as tf
# 创建模型
def create_model():
model = tf.keras.models.Sequential([
# 第1个卷积层
tf.keras.layers.Conv2D(32, (3, 3), activation='relu', input_shape=(28, 28, 1)),
tf.keras.layers.MaxPooling2D((2, 2)),
# 第2个卷积层
tf.keras.layers.Conv2D(64, (3, 3), activation='relu'),
tf.keras.layers.MaxPooling2D((2, 2)),
# 第3个卷积层
tf.keras.layers.Conv2D(64, (3, 3), activation='relu'),
# 展平输出层
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(10, activation='softmax')
])
return model
# 编译模型
def compile_model(model):
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
# 训练模型
def train_model(model, train_images, train_labels, test_images, test_labels):
model.fit(train_images, train_labels, epochs=10,
validation_data=(test_images, test_labels))
```
这是一个简单的卷积神经网络模型,可以用来对 MNIST 数据集进行分类。注意,这只是一个示例代码,实际使用中需要根据实际情况进行修改和优化。