python卷积神经网络代码
时间: 2023-10-12 17:05:01 浏览: 100
以下是使用Keras框架实现卷积神经网络的示例代码:
```python
from keras.models import Sequential
from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
# 定义模型
model = Sequential()
# 添加卷积层
model.add(Conv2D(filters=32, kernel_size=(3, 3), activation='relu', input_shape=(28, 28, 1)))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(filters=64, kernel_size=(3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(filters=128, kernel_size=(3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
# 添加全连接层
model.add(Flatten())
model.add(Dense(units=128, activation='relu'))
model.add(Dense(units=10, activation='softmax'))
# 编译模型
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
# 训练模型
model.fit(x_train, y_train, epochs=10, batch_size=32, validation_data=(x_test, y_test))
```
在上面的示例代码中,我们定义了一个包含3个卷积层和一个全连接层的卷积神经网络模型。其中,卷积层通过`Conv2D`类实现,池化层通过`MaxPooling2D`类实现。最后,我们使用`Dense`类添加全连接层,再通过`compile`方法编译模型,并使用`fit`方法训练模型。
阅读全文