当前位置:   article > 正文

深度学习入门demo:样本数据为y=0.7x+3.1函数的模型训练过程_模型训练 样本数据

模型训练 样本数据

案例简介:样本数据为y=0.7x+3.1函数的样本随机点;通过建立模型、训练优化最后预测出这些样本可以用哪个模型来保存,为之后的样本扩大提供了基础。

具体代码如下

  1. # -*- coding: utf-8 -*-
  2. """
  3. Created on Thu Jun 4 17:58:43 2020
  4. @author: Dell
  5. """
  6. #深度学习:根据样本数据,构建一个足够准确的模型
  7. import tensorflow as tf
  8. import numpy as np
  9. x1=np.random.rand(300)#随机生成300个点
  10. y1=x1*0.7+3.1#这为样本数据
  11. #构建、训练模型
  12. b=tf.Variable(0.)
  13. k=tf.Variable(0.)
  14. y=k*x1+b#构建的模型
  15. #定义二次代阶函数 使模型的参数尽量准确
  16. loss=tf.reduce_mean(tf.square(y1-y))#使真实值与模型预测值足够接近
  17. #定义一个梯度下降法训练优化器
  18. optimizer=tf.train.GradientDescentOptimizer(0.3)
  19. #最小化上头定义的待代阶函数
  20. train=optimizer.minimize(loss)
  21. #初始化变量
  22. initial=tf.global_variables_initializer()
  23. with tf.Session() as sess:
  24. sess.run(initial)
  25. for count in range(402):
  26. sess.run(train)
  27. if count%100==0:
  28. print(count,":",sess.run([k,b]))
  29. if count==401:
  30. print(count,"_end:",sess.run([k,b]))

结果分析:

具体结果如下图,一共训练了402次,可以看出最后的k与b已经接近样本数据的参数。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/955439
推荐阅读
相关标签
  

闽ICP备14008679号