Skip to content

Latest commit

 

History

History
44 lines (34 loc) · 3.04 KB

File metadata and controls

44 lines (34 loc) · 3.04 KB

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)