Keras 是一个高级神经网络 API,运行在 TensorFlow、CNTK 或 Theano 上。它设计用于加速实验和简化深度学习。
特点
- 用户友好:Keras 提供了简洁的 API,使得构建和训练神经网络变得非常容易。
- 模块化:Keras 允许用户自定义网络架构,并可以与 TensorFlow、CNTK 或 Theano 的底层功能无缝集成。
- 可扩展性:Keras 支持多种预训练模型,并可以轻松扩展以适应不同的任务。
快速入门
假设你已经安装了 Keras,以下是一个简单的例子:
from keras.models import Sequential
from keras.layers import Dense
model = Sequential()
model.add(Dense(12, input_dim=8, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit(X_train, y_train, epochs=150, batch_size=10)
资源
更多关于 Keras 的信息,可以访问 Keras 官方文档。
Keras Logo