3-Tensorflow-demo_03-使用TF实现一元线性回归问题

本文通过使用TensorFlow实现了一元线性回归模型的构建与训练过程,详细展示了如何利用占位符、变量、正向传播、损失计算及优化器进行模型训练,最后通过可视化展示训练结果。
部署运行你感兴趣的模型镜像


import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt

"""
模拟一元线性回归数据 ,请大家用numpy自己生成。
"""
if __name__ == '__main__':
    with tf.Graph().as_default():
        # 一、构建模型图
        # 1、构建输入的占位符
        input_x = None
        input_y = None

        # 2、构建变量
        w = None
        b = None

        # 3、正向传播,得到预测值。
        y_pred = None
        # 4、计算模型损失
        loss = None
        # 5、定义优化器。(含义:使用梯度下降的方式求解让损失函数最小的模型参数)
        optimizer = tf.train.GradientDescentOptimizer(learning_rate=0.01)
        train_opt = optimizer.minimize(loss=loss)

        # 二、构建会话
        with tf.Session() as sess:
            # a、变量初始化
            pass
            # b、加载数据(训练的数据生成)
            train_x = None
            train_y = None
            # c、模型训练
            for e in range(1, 200):
                # 执行模型训练操作
                feed = {input_x: train_x, input_y: train_y}
                _, train_loss = sess.run([train_opt, loss], feed_dict=feed)
                print('Epoch:{} - Train Loss:{:.5f}'.format(e, train_loss))
            # d、使用训练数据,得到该数据的预测值,并做一个可视化操作。
            pass
D:\Anaconda\python.exe D:/AI20/HJZ/04-深度学习/2-TensorFlow基础/tf_基础代码/01_01Graph和Session.py
Tensor("add:0", shape=(3, 5), dtype=float32) Tensor("add_1:0", shape=(5, 3), dtype=float32) Tensor("MatMul:0", shape=(3, 3), dtype=float32)
2019-11-30 21:39:59.231831: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
<tensorflow.python.client.session.InteractiveSession object at 0x000001D80CAE3C50>
[[ 2.385983  3.385983  4.385983]
 [ 3.385983  5.385983  5.385983]
 [43.385983  3.385983  2.385983]
 [ 1.385983  3.385983  4.385983]
 [ 4.385983  4.385983  4.385983]]
[[151.05128   30.45549   32.843945]
 [157.43454   47.838726  53.22718 ]
 [192.84169   38.24591   41.634365]]

Process finished with exit code 0

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值