Tensorflow
一、tensorflow 入门
import numpy as np
import tensorflow as tf
w = tf.Variable(0, dtype=tf.float32)
# cost = tf.add(tf.add(w**2, tf.multiply(-10., w)), 25)
cost = w**2 - 10 * w + 25 # TensorFlow会自动转换这些运算符!
train = tf.train.GradientDescentOptimizer(0.01).minimize(cost) # 学习率 和 成本函数
init = tf.global_variables_initializer()
session = tf.Session()
session.run(init) # 只是执行了第二行代码!!
print(session.run(w))
0.0
session.run(train) # tensorflow 会根据 前向传播 自动生成 反向传播操作!
print(session.run(w))
0.099999994
for i in range(1000):
session.run(train)
print(session.run(w))
4.9999886
二、placehold 占位符功能
x = tf.placeholder(tf.float32, [3, 1])
# cost = w**2 - 10 * w + 25
cost = x[0][0]*w**2 + x[1][0]*w + w[2][0]
coefficients = np.array([[1.], [-10.], [25.]])
session.run(train, feed_dict={x:coefficients})
for i in range(1000):
session.run(train, feed_dict={x:coefficients}) # 每次训练自动填充掉x的占位符!
print(session.run(w))
coefficients = np.array([[1.], [-20], [100.]])
for i in range(1000):
session.run(train, feed_dict={x: coefficients})
print(session.run(w))
4.9999886
三、重温代码
import numpy as np
import tensorflow as tf
coefficients = np.array([[1], [-20], [25]])
w = tf.Variable([0], dtype=tf.float32)
x = tf.placeholder(tf.float32, [3, 1])
cost = x[0][0]*w**2 + x[1][0] * w + x[2][0]
train = tf.train.GradientDescentOptimizer(0.01).minimize(cost)
init = tf.global_variables_initializer()
session.run(init)
print(session.run(w))
for i in range(1000):
session.run(train, feed_dict={x: coefficients})
print(session.run(w))
[0.]
[9.999977]