线性回归模型——TensorFlow实现

 看着书上稍微写了一下代码,代码还完全特别理解,之后再好好仔细阅读

import numpy as np
import tensorflow as tf
#2.0的tensorflow没有placeholder板块,只有把2.0的版本降低到1.0的版本,才可以使用placeholder,此处为降低版本的代码
tf = tf.compat.v1
tf.disable_v2_behavior()
from tensorflow.examples.tutorials.mnist import input_data
wb=[]
X_data=np.random.randn(2000,3)
w_real=[0.3,0.5,0.1]
b_real=-0.2

noise=np.random.randn(1,2000)*0.1
y_data=np.matmul(X_data,w_real)+noise+b_real

X=tf.placeholder(tf.float32,shape=(None,3))
y=tf.placeholder(tf.float32,shape=None)
w=tf.Variable([[0,0,0]],dtype=tf.float32)
b=tf.Variable(0,dtype=tf.float32)
y_pred=tf.matmul(w,tf.transpose(X))+b

loss=tf.reduce_mean(tf.square(y-y_pred))
optimizer=tf.train.GradientDescentOptimizer(learning_rate=0.01)
train_op=optimizer.minimize(loss)

with tf.Session() as sess:
    tf.global_variables_initializer().run()
    for step in range(10):
        sess.run(train_op,feed_dict={X:X_data,y:y_data})
        if (step%5==0):
            print(step,sess.run([w,b]))
            wb.append(sess.run([w,b]))

    print(10,sess.run([w,b]))

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值