Keras 是一个高级神经网络 API,可以运行在 TensorFlow、CNTK 或 Theano 后端之上。本文将详细介绍如何使用 Keras API 进行模型训练。
安装和导入
首先,确保你已经安装了 Keras。可以使用以下命令进行安装:
pip install keras
然后,在 Python 中导入 Keras:
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation
创建模型
创建一个简单的全连接神经网络模型:
model = Sequential()
model.add(Dense(128, input_dim=64, activation='relu'))
model.add(Dropout(0.2))
model.add(Dense(1, activation='sigmoid'))
编译模型
在训练模型之前,需要编译它:
model.compile(loss='binary_crossentropy',
optimizer='adam',
metrics=['accuracy'])
准备数据
在训练模型之前,需要准备数据。以下是一个使用 NumPy 生成随机数据的示例:
import numpy as np
# 生成随机数据
x_train = np.random.random((1000, 64))
y_train = np.random.randint(2, size=(1000, 1))
x_test = np.random.random((200, 64))
y_test = np.random.randint(2, size=(200, 1))
训练模型
使用以下命令开始训练模型:
model.fit(x_train, y_train, epochs=10, batch_size=32)
评估模型
使用测试数据评估模型:
score = model.evaluate(x_test, y_test, batch_size=32)
print('Test loss:', score[0])
print('Test accuracy:', score[1])
扩展阅读
想了解更多关于 Keras 的信息?请访问我们的 Keras 教程。
图片示例
神经网络结构
中心化的神经网络结构: