写一个简单的深度学习python代码
时间: 2024-06-05 15:09:12 浏览: 88
好的,我可以给你一个简单的深度学习代码示例:
import numpy as np
import tensorflow as tf
# 准备样本数据
train_x = np.random.rand(100, 3)
train_y = 2 * train_x[:,0] + train_x[:,1] - 3 * train_x[:,2] + 1
# 建立模型
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(10, activation='relu', input_shape=(3,)),
tf.keras.layers.Dense(1)
])
# 编译模型
model.compile(optimizer='adam',
loss='mse',
metrics=['mae'])
# 训练模型
history = model.fit(train_x, train_y, epochs=100, verbose=False)
# 预测结果
test_x = np.random.rand(1, 3)
test_y = 2 * test_x[:,0] + test_x[:,1] - 3 * test_x[:,2] + 1
predicted_y = model.predict(test_x)
print('预测结果:', predicted_y)
print('真实结果:', test_y)
阅读全文