US-Stock-Prediction-Using-ML-And-Spark

KO | EN

gitlite — search

#typescript #ai-agents #ai #dsh-plugin #deepseek-harness #open-source #cli #claude-code #codex #developer-tools #react #windows

US-Stock-Prediction-Using-ML-And-Spark

★ 59 Open GitHub ↗

Predict stock price based on financial news feeds

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Stock Price Prediction via Financial News Sentiment Analysis

#### Data Source:
1. WSJ News 
 - Full-attribute Wall St. Journal News, including news published timestamp, keywords list, news headline, news body content, and news url. All in JSON file format.
2. Reuters News 
 - Same as WSJ news data.
3. Intra-Day Stock Price 
 - By-minute stock price scrapped from Google Stock API.
 - Use tickers from three major U.S. Enquity Stock Trade Exchanges: Nasdaq, NYSE, AMEX.
 - Scrapped data only contains the records that have price changes, meaning that if one stock does not have any price change in one hour, there will be no by-minute record within that hour. 
4. meta/alias2Ticker.json
 - This file is collected from running queries with stock tickers to CityFalcon (https://www.cityfalcon.com/). It is still under construction and cleaning for better matching accuracy.
 - JSON schema: {"alias":"american airlines group","ticker":"AAL"}
 - use stopword.txt to delete tickers and aliases that show as common English words.
5. meta/tickerInfo.json
 - JSON schema: {"ticker": "ABC", "sector": "Some_Sector", "category": "Some_Cat", "group": "Some_Group"}
 - Includes 9 sectors, 31 categories, 212 groups

#### Data Pipeline
<img width="659" alt="data pipeline" src="https://user-images.githubusercontent.com/15644582/37743533-e6e4f3e8-2d40-11e8-91c2-7a7864f3f7b3.png">

#### Actual Data Flow
<img width="616" alt="dataflow" src="https://user-images.githubusercontent.com/15644582/37743565-0a5f67d6-2d41-11e8-9765-9ffab9cf4fa6.png">

#### Predicting Result
We found that a Linear Regression model could get quite good result. The comparision of different reacting time shows that the stock is sensitive to the financial press. The 7 minutes reacting time has highest accuracy among others.
<img width="399" alt="screen shot 2018-03-21 at 8 02 36 pm" src="https://user-images.githubusercontent.com/15644582/37743916-d8cf5198-2d42-11e8-9040-3abb761cbe34.png">

Also, here is a sample of HeatMap of investment table, x-axis is company name represented by stock ticker, and y-axis is date of month. Well konwn companies like Google or Apple were on the press very often, but others like AMD showed up just once at that month.

#### Reference:
1. Git: http://www.vogella.com/tutorials/Git/article.html
2. UDFs: https://jaceklaskowski.gitbooks.io/mastering-apache-spark/spark-sql-udfs.html
3. Dataset: https://www.balabit.com/blog/spark-scala-dataset-tutorial/
4. DataFrames: https://docs.databricks.com/spark/latest/dataframes-datasets/introduction-to-dataframes-scala.html  
5. Word2Vec Sentiments: https://github.com/linanqiu/word2vec-sentiments

## To-Do:
#### Improvement:
1. Stocks that do not have contineuous by-minute price record needs better simulation. Currently, the simulation is done by having all the missing records be the one that appear the latest. We need to mock up the missing price in a linear fashion between two price record.
2. Inspect the alias2ticker json file: 
 - To eliminate confusing aliases.  
 - To remove aliases that referring to non-US traded ticker symbols.   
 - To add potential match-able aliases.  
 - To add ticker symbols that constains .[dot]  
 - To add alias-ticker pairs for missing tickers.
3. We have not tried the Deep Learning approaches; however, past researches have shown steep improvement with DL.
4. Automate the data ETL process.
5. Render real-time result on Tableau.

#### Experiment:
1. Currently, the target is computed from the average price of the _open_, _close_, _highest_, and _lowest_. We have not take the _volume_ into the training. We should also try to work out variations on target value.
2. Consult with Financial Engineering researchers on constructing models that have more factors in consideration.

US-Stock-Prediction-Using-ML-And-Spark

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README