Keras 是一个高级神经网络 API,能够以用户友好的方式定义和训练神经网络。以下是 Keras 框架的一些基本教程内容。
快速开始
安装 Keras:首先确保你已经安装了 TensorFlow,Keras 是 TensorFlow 的一部分。
pip install tensorflow
导入 Keras:
from keras.models import Sequential from keras.layers import Dense, Activation
构建模型:
model = Sequential() model.add(Dense(64, input_dim=10)) model.add(Activation('relu')) model.add(Dense(1, activation='sigmoid'))
数据预处理
在开始训练之前,需要对数据进行预处理。
导入数据:
import numpy as np x_train = np.random.random((1000, 20)) y_train = np.random.randint(2, size=(1000, 1))
编译模型:
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
训练模型:
model.fit(x_train, y_train, epochs=20, batch_size=32)
模型评估
完成训练后,可以对模型进行评估。
score = model.evaluate(x_train, y_train, batch_size=32)
print('Test loss:', score[0])
print('Test accuracy:', score[1])
扩展阅读
想要了解更多关于 Keras 的信息,可以阅读 Keras 官方文档。
TensorFlow Logo