下面我将为您详细讲解在Win10下搭建Python3.5.2和TensorFlow环境的步骤,并附带两个示例说明。
python
命令测试是否安装成功。如果出现Python版本号,则说明Python3.5.2的环境已经成功搭建起来了。pip install tensorflow==1.15
python
命令,并使用以下代码进行测试:import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
如果输出了“Hello, TensorFlow!”,则说明TensorFlow环境已经成功安装好了。
下面我们通过一个简单的示例来使用TensorFlow进行线性回归。
import tensorflow as tf
import numpy as np
# 生成训练数据集
x_data = np.random.rand(100).astype(np.float32)
y_data = x_data * 0.1 + 0.3
# 构造模型
weights = tf.Variable(tf.random_uniform([1], -1.0, 1.0))
bias = tf.Variable(tf.zeros([1]))
y = weights * x_data + bias
# 定义损失函数
loss = tf.reduce_mean(tf.square(y - y_data))
# 定义优化器
optimizer = tf.train.GradientDescentOptimizer(0.5)
train = optimizer.minimize(loss)
# 初始化变量
init = tf.global_variables_initializer()
# 启动Session
sess = tf.Session()
sess.run(init)
# 开始训练
for step in range(201):
sess.run(train)
if step % 20 == 0:
print(step, sess.run(weights), sess.run(bias))
上述代码中,我们首先构造了一个线性模型,然后随机生成了100个训练数据,最终使用随机梯度下降算法进行优化,使得模型能够拟合训练数据。
下面我们通过一个示例来使用TensorFlow进行图像分类。
import tensorflow as tf
from tensorflow import keras
import numpy as np
# 下载并载入MNIST数据集
mnist = keras.datasets.mnist
(train_images, train_labels), (test_images, test_labels) = mnist.load_data()
# 对数据进行预处理
train_images = train_images / 255.0
test_images = test_images / 255.0
# 构造模型
model = keras.Sequential([
keras.layers.Flatten(input_shape=(28, 28)),
keras.layers.Dense(128, activation='relu'),
keras.layers.Dense(10, activation='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)
上述代码中,我们使用了Keras高层次的API,来构造一个简单的全连接神经网络模型,用于对MNIST手写数字进行识别。我们通过训练和测试的日志,可以看到模型在测试数据集上的识别准确率高达98%以上。
本文链接:http://task.lmcjl.com/news/5029.html