tensorflow实现线性回归例子
一、环境
python3.5版本、pycharm2020、TensorFlow1.14.1
(关于pycharm版本没有什么关系,TensorFlow的版本最好是1.几版本,否则可能存在运行不了的情况)
二、要求和实现
主要描述tensorflow实现线性回归的流程
三、实现流程
线性回归是利用数理统计中回归分析,来确定两种或两种以上变量间相互依赖的定量关系的一种统计分析方法。表达形式为y = w’x+e,e为误差服从均值为0的正态分布。
在建模的时候一般需要绘制散点图。
用TensorFlow实验线性回归的话,首先,我们需要样本数据 - train_x,train_y、随机生成变量w,b。接着可以创建线性模型 y=wx+b。我们还需要创建损失函数cost,再对函数用梯度下降算法优化损失函数。完成之后,创建一个session会话
并对变量进行初始化,开始迭代训练。还可以计算模型在数据集上的损失、最终的Loss。最后画出拟合曲线,更好的进行观察。
四、代码详解
import tensorflow as tf
import numpy
import matplotlib.pyplot as plt
rng = numpy.random
# Parameters
learning_rate = 0.01#设置梯度下降算法的学习率,一般为0~1之间比较小的值
training_epochs = 1000 #设置迭代次数
display_step = 50 #每迭代100次输出一次loss
# Training Data
#训练数据,生成生成样本数据train_x,train_y。asarray可以将元组,列表,元组列表,列表元组转化成ndarray对象。
train_X = numpy.asarray([3.3,4.4,5.5,6.71,6.93,4.168,9.779,6.182,7.59,2.167,
7.042,10.791,5.313,7.997,5.654,9.27,