Tensorflow 实现线性回归

import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
import tensorflow as tf 
import numpy as np
import matplotlib.pyplot as plt

def normalize(X):
    mean = np.mean(X)
    std = np.std(X)
    X = (X-mean)/std
    return X

boston = tf.contrib.learn.datasets.load_dataset('boston')
X_train = boston.data[:,5]
Y_train = boston.target
n_samples = len(X_train)

X = tf.placeholder(tf.float32, name='X')
Y = tf.placeholder(tf.float32, name='Y')

b=tf.Variable(0.0)
w=tf.Variable(0.0)

Y_hat = X * w + b

loss = tf.square(Y-Y_hat, name='loss')

optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.01).minimize(loss)

init = tf.global_variables_initializer()
total = []
with tf.Session() as sess:
    sess.run(init)

    for i in range(1000):
        total_loss = 0
        for x,y in zip(X_train,Y_train):
            _, l = sess.run([optimizer,loss], feed_dict={X:x,Y:y})
            total_loss += l
    
        total.append(total_loss/n_samples)
        print('Epoch {0}: Loss {1}'.format(i,total_loss/n_samples))

    b_value, w_value = sess.run([b, w])

Y_pred = X_train * w_value + b_value
print('Done')

plt.plot(X_train,Y_train,'bo',label='Real Data')
plt.plot(X_train,Y_pred,'r', label='Predicted Data')
plt.legend()
plt.show()

plt.plot(total)
plt.show()