1         #線性回歸：用線性模型y=Wx+b擬合sin
 2         import numpy as np
 3         import matplotlib.pyplot as plt
 4         import tensorflow as tf
 5         
 6         #數據，標簽
 7         x_data  = np.linspace(-2*np.pi,2*np.pi,300)
 8         noise = np.random.normal(-0.01,0.05,x_data.shape)
 9         y_label = np.sin(x_data) + noise
 
10         plt.rcParams[‘font.sans-serif‘]=[‘FangSong‘] # 用來正常顯示中文標簽
11         plt.rcParams[‘axes.unicode_minus‘]=False# 用來正常顯示負號
12         plt.title(‘線性模型y=Wx+b擬合sin‘)
13         plt.legend()
14         plt.grid(True)
15         plt.plot(x_data, y_label, ‘b.‘, label=‘測試數據‘)
16         
17         #
 
靜態圖定義
18         mg = tf.Graph()
19         with mg.as_default():
20             #圖輸入
21             X = tf.placeholder("float")
22             Y = tf.placeholder("float")
23             
24             #訓練權重w,b
25             W = tf.Variable(np.random.randn(), name="weight")
26             b = tf.Variable(np.random.randn(), name="
 
bias")
27             
28             #線性模型y=Wx+b
29             pred = tf.add(tf.multiply(X, W), b)
30             
31             #損失函數:使用樣本方差
32             cost = tf.reduce_sum(tf.pow(pred-Y, 2)) / (len(x_data)-1)
33             
34             #使用梯度下降法優化
35             optimizer = tf.train.GradientDescentOptimizer(0.01).minimize(cost)
36             
37             #初始化圖變量
38             init = tf.group(tf.global_variables_initializer(),
39                             tf.local_variables_initializer())
40             
41         with tf.Session(graph=mg) as sess:
42             sess.run(init)
43             for epoch in range(500):
44                 for (x, y) in zip(x_data, y_label):
45                     sess.run(optimizer, feed_dict={X: x, Y: y})
46                 if (epoch+1) % 10 == 0:
47                     c = sess.run(cost, feed_dict={X: x_data, Y:y_label})
48                     print(‘epoch={:} cost={:0.6f} W={:0.6f} b={:0.6f}‘.format(epoch+1,c,sess.run(W),sess.run(b)))
49             training_cost = sess.run(cost, feed_dict={X: x_data, Y: y_label})
50             print(‘訓練結果 cost={:0.6f} W={:0.6f} b={:0.6f}‘.format(training_cost,sess.run(W),sess.run(b)))
51             plt.plot(x_data, sess.run(W) * x_data + sess.run(b),‘r--‘ , label=‘擬合數據‘)
52             plt.legend()
53         plt.show()

使用tensorflow實現最簡單的線性回歸算法