TensorFlow 实战多元线性回归问题

TensorFlow 实战

多元线性回归问题

本文小编将以一个具体例子来说明如何用 TensorFlow 来实现多元线性回归问题。我们以波士顿房价预测为例进行讲解。

1、 情景引入

波士顿房价数据集包括506个样本,每个样本包括12个特征变量和该地区的平均房价。房价显然和多个特征变量是相关的,因此需要选择多个特征变量来建立线性方程,这就是典型的多变量线性回归问题,下面是多元线性回归模型公式:y=w0+w1x1+w2x2+...+wn*xn+b
下面是波士顿房价的数据集部分展示,后面我们将使用数据集进行模型实现。image.png

2、读取数据并归一化

通过pandas读取数据文件,列出统计概述。

 
 
  1. %matplotlib inline

  2. import matplotlib.pyplot as plt

  3. import numpy as np

  4. import tensorflow as tf

  5. import pandas as pd


  6. #读取数据文件

  7. df = pd.read_csv("./boston.csv",header=0)

  8. #显示数据摘要描述信息

  9. print(df.describe())

image.png

 
 
  1. #对数据归一化到0-1之间

  2. for i in range(12):

  3.    df[:,i]=(df[:,i]-df[:,i].min())/(df[:,i].max()-df[:,i].min())


  4. #x_data为归一化后的前12列特征数据

  5. x_data = df[:,:12]

  6. #y_data为最后一列标签数据

  7. y_data = df[:12]

3、模型定义

首先定义特征数据和标签数据的占位符,shape中None表示行的数量未知,在实际训练时决定一次代入多少行样本,从一个样本的随机SDG到批量SDG都可以。

x = tf.placeholder(tf.float32,[None,12],name="x")y = tf.placeholder(tf.float32,[None,1],name="y")

定义模型函数image.png

4、训练模型

设置训练超参数

#迭代次数train_epochs = 50#学习率learning_rate = 0.01

定义均方差损失函数

with tf.name_scope("LossFunction"):    loss_function = tf.reduce_mean(tf.pow(y-pred,2)) #均方误差

选择优化器,初始化一个GradientDescentOptimizer

#梯度下降优化器optimizer = tf.train.GradientDescentOptimizer(learning_rate).minimize(loss_function)

5、创建会话

声明会话

sess = tf.Session()

变量初始化

init = tf.global_variables_initializer()sess.run(init)

6、迭代训练

模型训练阶段,设置迭代次数,每次通过将样本逐个输入模型,进行梯度下降优化操作,每轮迭代后,绘制出模型曲线image.png下图为最后的训练结果image.png

7、训练结果的可视化

image.png

plt.plot(loss_list)

image.png

8、TensorBoard 可视化

声明会话

sess = tf.Session()#定义初始化变量的操作init = tf.global_variables_initializer()

为 TensorBoard 可视化准备数据

#设置日志存储目录logdir = 'd:/log'sum_Loss = tf.summary.scalar("loss",loss_function)merged = tf.summary.merge_all()

创建摘要的文件写入器

sess.run(init)writer = tf.summary.FileWwriter(logdir,sess.graph)

查看计算图image.pngTensorBoard查看lossimage.png好啦,最后赶快动手尝试一下属于你们的第一个多元回归模型的建立吧!

本文来自读者投稿,欢迎大家点击下面链接进行投稿:
欢迎投稿


---------End---------

image.png

猜你喜欢

转载自blog.51cto.com/15064628/2599799