📅  最后修改于: 2023-12-03 15:02:30.437000             🧑  作者: Mango
Keras是一个高级神经网络API,以TensorFlow,CNTK或Theano作为后端。它简化了许多常见的深度学习任务,例如图像分类,自然语言处理,对象检测等,使得深度学习更加易于上手。
Keras的优点:
下面是一个Keras构建CNN模型的示例代码片段:
from keras.models import Sequential
from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
# 创建序列模型
model = Sequential()
# 添加两个卷积层,使用32个过滤器,每个过滤器大小为(3,3),使用ReLU激活函数
model.add(Conv2D(32, (3,3), input_shape=(28,28,1), activation='relu'))
model.add(Conv2D(32, (3,3), activation='relu'))
# 添加2x2最大池化层
model.add(MaxPooling2D(pool_size=(2,2)))
# 添加Flatten层,将卷积层输出压缩成一维向量
model.add(Flatten())
# 添加全连接层,输出为128个单元,使用ReLU作为激活函数
model.add(Dense(128, activation='relu'))
# 添加输出层,输出单元为10,使用softmax激活函数
model.add(Dense(10, activation='softmax'))
# 输出模型概述
model.summary()
以上代码创建了一个简单的CNN模型,包括两个卷积层,一个最大池化层,一个Flatten层,一个全连接层和一个输出层。通过使用Keras简单的API,我们可以快速构建出一个功能强大的神经网络模型。
参考文献: