TensorFlow 的 Keras API 是一个高级神经网络库,它提供了构建和训练神经网络所需的所有工具。Keras API 非常易于使用,适合快速原型制作和实验。
快速开始
安装
首先,您需要安装 TensorFlow:
pip install tensorflow
创建模型
创建一个简单的全连接神经网络:
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
model = Sequential()
model.add(Dense(64, activation='relu', input_shape=(100,)))
model.add(Dense(1, activation='sigmoid'))
训练模型
使用一些数据来训练模型:
model.compile(optimizer='adam',
loss='binary_crossentropy',
metrics=['accuracy'])
model.fit(data, labels, epochs=10, batch_size=32)
评估模型
使用测试数据评估模型:
test_loss, test_acc = model.evaluate(test_data, test_labels, verbose=2)
print('\nTest accuracy:', test_acc)
保存和加载模型
保存模型:
model.save('my_model.h5')
加载模型:
from tensorflow.keras.models import load_model
new_model = load_model('my_model.h5')
更多信息和教程,请访问 TensorFlow 官方文档。
Keras API 示例