帮我用python写一个人工智能的代码

当然可以，我可以为您提供一个简单的示例代码：

import tensorflow as tf

# 创建一个简单的神经网络模型
model = tf.keras.Sequential([
    tf.keras.layers.Dense(64, activation='relu', input_shape=(784,)),
    tf.keras.layers.Dense(10, activation='softmax')
])

# 编译模型
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

# 训练模型
model.fit(x_train, y_train, epochs=5)

# 使用模型进行预测
predictions = model.predict(x_test)

这是一个使用 TensorFlow 库创建神经网络模型的示例代码，您可以根据自己的需求进行修改和扩展。