你好 帮我写一个python的深度学习的代码

嗨，你好！我可以为你提供一段python深度学习代码：import tensorflow as tf
from tensorflow import keras# 定义神经网络模型
model = keras.Sequential([
keras.layers.Flatten(input_shape=(28, 28)), # 输入层，把28x28的数据拉成一维
keras.layers.Dense(128, activation='relu'), # 隐藏层，128个神经元，激活函数为relu
keras.layers.Dense(10, activation='softmax') # 输出层，10个神经元，激活函数为softmax
])# 编译模型
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])# 训练模型
model.fit(train_images, train_labels, epochs=5)# 评估模型
test_loss, test_acc = model.evaluate(test_images, test_labels)
print('Test accuracy:', test_acc)