第三章 初识TensorFlow

如果已知两位学生的成绩

90 * w1 + 80 * w2 + 70 * w3 = 85

98 * w1 + 95 * w2 + 87 * w3 = 96

两个方程式解三个未知数依然无法求解

• 代码：

import tensorflow as tf

x1 = tf.placeholder(dtype=tf.float32)

x2 = tf.placeholder(dtype=tf.float32)

x3 = tf.placeholder(dtype=tf.float32)

yTrain = tf.placeholder(dtype=tf.float32)

w1 = tf.Variable(0.1, dtype=tf.float32)

w2 = tf.Variable(0.1, dtype=tf.float32)

w3 = tf.Variable(0.1, dtype=tf.float32)

n1 = x1 * w1

n2 = x2 * w2

n3 = x3 * w3

y = n1 + n2 + n3

loss = tf.abs(y - yTrain)

optimizer = tf.train.RMSPropOptimizer(0.001)

train = optimizer.minimize(loss)

sess = tf.Session()

init = tf.global_variables_initializer()

sess.run(init)

for i in range(5000):

                result = sess.run([train, x1, x2, x3, w1, w2, w3, y, yTrain, loss], feed_dict={x1: 90, x2: 80, x3: 70, yTrain: 85})

        print(result)

                result = sess.run([train, x1, x2, x3, w1, w2, w3, y, yTrain, loss], feed_dict={x1: 98, x2: 95, x3: 87, yTrain: 96})

        print(result)

• 代码说明：

神经网络在投入使用前，都要经过训练（train）的过程才能有准确的输出

w1、w2、w3已经非常接近于预期的0.6、0.3、0.1，y也非常接近于目标值

• 运行结果：

1、安装完成

2、调试完成