如何使用 Python 的 Keras 库进行数据训练?
步骤:
- 导入必要的库
import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense, Dropout
- 加载数据
(x_train, y_train), (x_test, y_test) = mnist.load_data()
- 创建模型
model = Sequential()
model.add(Dense(128, activation='relu', input_shape=(28, 28, 1)))
model.add(Dropout(0.2))
model.add(Dense(10, activation='softmax'))
- 编译模型
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
- 训练模型
model.fit(x_train, y_train, epochs=5)
- 评估模型
loss, accuracy = model.evaluate(x_test, y_test)
print('Loss:', loss)
print('Accuracy:', accuracy)
完整代码:
import keras
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense, Dropout
# 加载数据
(x_train, y_train), (x_test, y_test) = mnist.load_data()
# 创建模型
model = Sequential()
model.add(Dense(128, activation='relu', input_shape=(28, 28, 1)))
model.add(Dropout(0.2))
model.add(Dense(10, activation='softmax'))
# 编译模型
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
# 训练模型
model.fit(x_train, y_train, epochs=5)
# 评估模型
loss, accuracy = model.evaluate(x_test, y_test)
print('Loss:', loss)
print('Accuracy:', accuracy)
注意:
-
mnist.load_data()
函数加载 MNIST 数据集,它包含 70,000 个 28x28 像素的彩色图像,每个图像对应一个数字。 -
model.compile()
函数用于指定模型的训练参数,包括优化器、损失函数和评估指标。 -
model.fit()
函数用于训练模型,它将使用给定的训练数据训练模型。 -
model.evaluate()
函数用于评估模型,它将使用给定的测试数据评估模型的准确性。