Skip to content
Switch branches/tags
Go to file

Latest commit


Git stats


Failed to load latest commit information.
Latest commit message
Commit time
Nov 29, 2018
Nov 29, 2018


In Wall Street, the global financial center, the proportion of investment in artificial intelligence has increased gradually since the 2008 financial crisis (Source: Bloomberg) In addition, artificial intelligence can make reasonable decisions because it does not pay for the inefficiency of investment sentiment in determining whether to invest and scale.(E.g, The one AI system at the time of the global financial crisis in 2008 recorded a 681% )


We will use each of historical stock price dataset from the Reuters. The dataset contains the raw time-series data(Open, High, Low, Close, and Volume).

Details of Dataset and Models

  • There are 10 stocks dataset(From 2015 ~ From 2017) and KOSPI index.

  • We use 10 stocks,KOSPI dataset not only 2015/2016(Jan-Dec) as train/Validation dateset, also 2016/2017(Jan-Dec) as test dateset.

  • We use XGBoost model. **Official website:

  • XGBoost and other ensemble models is one of learning methods to predict stock prices. Afterwards based on past market data, stocks (listed in KOSPI market) are subject to post-verification(back-testing) and real-time simulation investment.

  • We calculate rates of each stock of returns every at the end of each week.


  • This is an example which is how well train/predict.(For details, Please refer the code "")


  • XGboost (0.7)
  • numpy (1.15.1)
  • matplotlib (2.2.2)
  • scikit-learn (0.19.1)
  • Pandas (0.22.0)
  • Scipy (1.1.0)


Apache License 2.0


If you have any question, please contact Eunji Bang(

XAI Project

This work was supported by Institute for Information & Communications Technology Promotion(IITP) grant funded by the Korea government(MSIT) (No.2017-0-01779, A machine learning and statistical inference framework for explainable artificial intelligence)

  • Project Name : A machine learning and statistical inference framework for explainable artificial intelligence(의사결정 이유를 설명할 수 있는 인간 수준의 학습·추론 프레임워크 개발)

  • Managed by Ministry of Science and ICT/XAIC

  • Participated Affiliation : UNIST, Korea Univ., Yonsei Univ., KAIST, AItrics

  • Web Site :