作家
登录

自创数据集,用TensorFlow预测股票教程

作者: 来源: 2017-11-24 09:31:59 阅读 我要评论

Tech Neo技巧沙龙 | 11月25号,九州云/ZStack与您一路商量云时代收集界线治理实践


自创数据集,用TensorFlow猜测股票教程 (附代码)

STATWORX 团队近日大年夜 Google Finance API 中精选出了 S&P 500 数据,该数据集包含 S&P 500 的指数和股价信息。有了这些数据,他们就欲望能应用深度进修模型和 500 支成分股价猜测 S&P 500 指数。STATWORX 团队的数据集十分新鲜,但只是应用四个隐蔽层的全连接收集实现猜测,读者也可以下载该数据测验测验加倍优良的轮回神经收集。

本文异常合适初学者懂得若何应用 TensorFlow 构建根本的神经收集,它周全展示了构建一个 TensorFlow 模型所涉及的概念与模块。本文所应用的数据集可以直接下载,所以有必定基本的读者也可以测验测验应用更强的轮回神经收集处理这一类时序数据。

数据集地址:http://files.statworx.com/sp500.zip

导入和预处理数据

STATWORX 团队大年夜办事器爬取股票数据,并将它们保存为 csv 格局的文件。该数据集包含 n=41266 分钟的记录,范围大年夜 2017 年的 4 月到 8 月的 500 支股票和 S&P 500 指数,股票和股指典范围分布十分广。

  1. # Run initializer 
  2.  
  3. net.run(tf.global_variables_initializer()) 
  4.  
  5.  
  6.  
  7. # Setup interactive plot 
  8.  
  9. plt.ion() 
  10.  
  11. fig = plt.figure() 
  12.  
  13. ax1 = fig.add_subplot(111) 
  14.  
  15. line1, = ax1.plot(y_test) 
  16.  
  17. line2, = ax1.plot(y_test*0.5) 
  18.  
  19. plt.show() 
  20.  
  21.  
  22.  
  23. # Number of epochs and batch size 
  24.  
  25. epochs = 10 
  26.  
  27. batch_size = 256for e in range(epochs): 
  28.  
  29.  
  30.  
  31. # Shuffle training data 
  32.  
  33. shuffle_indices = np.random.permutation(np.arange(len(y_train))) 
  34.  
  35. X_train = X_train[shuffle_indices] 
  36.  
  37. y_train = y_train[shuffle_indices] 
  38.  
  39.  
  40.  
  41. # Minibatch training 
  42.  
  43. for i in range(0, len(y_train) // batch_size): 
  44.  
  45. start = i * batch_size 
  46.  
  47. batch_x = X_train[start:start + batch_size] 
  48.  
  49. batch_y = y_train[start:start + batch_size] 
  50.  
  51. # Run optimizer with batch 
  52.  
  53. net.run(opt, feed_dict={X: batch_x, Y: batch_y}) 
  54.  
  55.  
  56.  
  57. # Show progress 
  58.  
  59. if np.mod(i, 5) == 0: 
  60.  
  61. # Prediction 
  62.  
  63. pred = net.run(out, feed_dict={X: X_test}) 
  64.  
     1/6    1 2 3 4 5 6 下一页 尾页

      推荐阅读

      敏捷框架比较:Scrum vs Kanban vs Lean vs XP

    Tech Neo技巧沙龙 | 11月25号,九州云/ZStack与您一路商量云时代收集界线治理实践 在这篇文┞仿中, Alesia Krush将对四种最风行的敏捷开辟办法进行比较,给出了每种办法的优缺点。市场上有>>>详细阅读


    本文标题:自创数据集,用TensorFlow预测股票教程

    地址:http://www.17bianji.com/lsqh/39204.html

关键词: 探索发现

乐购科技部分新闻及文章转载自互联网,供读者交流和学习,若有涉及作者版权等问题请及时与我们联系,以便更正、删除或按规定办理。感谢所有提供资讯的网站,欢迎各类媒体与乐购科技进行文章共享合作。

网友点评
自媒体专栏

评论

热度

精彩导读
栏目ID=71的表不存在(操作类型=0)