Skip to content

SingingData/StockPerformanceClassification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

102 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Stock Performance Classification with a 1D CNN, Keras and Azure ML Workbench

Overview

We recently worked with a financial services partner to develop a model to predict the future stock market performance of public companies in categories where they invest. The goal was to use select text narrative sections from publicly available earnings release documents to predict and alert their analytics to investment opportunities and risks. We developed a deep learning model using a one-dimensional convolutional neural network (a 1D CNN) based on text extracted from public financial statements from these companies to make these predictions. We used Azure Machine Learning Workbench to explore the data and develop the model. We modeled using the Keras deep learning Python framework with a Theano backend. The results demonstrate how a deep learning model trained on text in earnings releases and other sources could provide a valuable signal to the investment decision maker.

This initial result suggests that that deep learning models trained on text in earnings releases and other sources could prove a viable mechanism to improve the quality of the information available to the person making the investment decision, particularly in avoiding investment losses. While the model needs to be improved with more samples, refinements of domain-specific vocabulary and text augmentation, this model suggests that providing this signal as another decision input for investmennt analyst would improve the efficiency of the firm’s analysis work.

The history of model training and testing is below, trained for 24 epochs. alt text

Model Architecture

This graphic depicts the model architecture. alt text

This graphic gives you a summary of this model at each level.
alt text

Azure ML Workbench

We built the solution on the Azure ML Workbench python environment. We found the following installs and upgrades were required.

Installs and Upgrades Required (Order is Important)

About

Keras 1D CNN on Azure ML Workbench to classify 4 week stock performance based on text in public earnings statements

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors