Need help with Tensorflow-for-stock-prediction?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

450 Stars 231 Forks 42 Commits 5 Opened issues


Use Tensorflow to run CNN for predict stock movement. Hope to find out which pattern will follow the price rising.

Services available


Need anything else?

Contributors list

# 145,303
38 commits

Tensorflow work for stock prediction

Use Tensorflow to run CNN for predict stock movement. Hope to find out which pattern will follow the price rising. Different implement codes are in separate folder.

Feature include daily close price, MA, KD, RSI, yearAvgPrice..... Detail described as below. This work is just an sample to demo deep learning. The result is not well estimated.

All of the work are done by using the same stock(2330 in Taiwan stock) which are collected from Please notice that this stock perform a stable rise these years. But our result get a little better than just random guess.

For Brief tutorial slider please check (Distributed Tensorflow & Stock prediction)

For Chinese outline slider please check HERE(中文簡介)


  • Tensorflow
  • yahoo_finance
  • numpy If you run DQNKDvalue, DQNcnnimage you'll need extra
  • matplotlib
  • cv2


The major work of this project. We feed data(yearline,monthline, closePrice) as image and use CNN to recognize their pattern. Some resources really help a lot while building DQN. The main difference is how to set the reward function and way to train Qnetwork. * *

File: - :use for making yearline img and closeprice img, and then build model. - :to build model which should be train for about 24hr. //run first - Test model by yearline.ipynb : There is one model exsit in savedyearr. The code create some img to test on that. - build a model by closeprice img and do evaluation.


exp 1

  • Training: 2011~2014 15-day image with only daily close price.
  • Testing : 2016/01 ~2016/08
  • Trading strategy: Reward=(Tomorrow's close price)-(today's close price) if predict buy. Negate the number while predicting sell.
  • Bad performance.

alt tag

exp 2

  • Trading strategy: sell while meet +10% profit or -5% loss.
  • Baseline: Considering of the rising stock price, the baseline is the average profit takes buying times into account.
  • We get more profit then baseline. And the testing file can be used for evaluating whether to believe in the model.

alt tag


The model is originated from RobRomijnders 's work used for time series data ( I've change some parameter and use different indicators to find out when to buy the stock is good.


Part 1. Indicators

Feature MA can drop the loss compare with RSI and ClosePrice at training step.

alt tag

Part 2. When To Buy

With 10 days MA5 as an instance. * Training data (2330train15) : 2001~2014 Instance labeled as 1 when it rising dramatically up to 15% in 90 days. And mix with 4 times instance labeled as 0 * Testing data (2330_test) : 2015/07 ~ 2016/08 MA5.

After running CNN_Classifier.ipynb, Result will be visualized.

alt tag


My implement is under close price. This could be change to other features like RSI,KD,MA....Or, use all of them. There is CNN code that could be edit to meet the requirement (size of batch).


Not work in closePrice. Better with other feature.


Use KD value picture to predict.

python //this call and build model.


  • Training: 2011~2014 15-day K value and D value image
  • Testing : 2015/07 ~2016/07
    alt tag

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.