26hzhang / StockPrediction

Plain Stock Close-Price Prediction via Graves LSTM RNNs
MIT License
194 stars 113 forks source link
deeplearning4j java lstm recurrent-neural-networks spark-dataframes stock-price-prediction

Plain Stock Prediction

Authour

Plain Stock Price Prediction via RNNs with Graves LSTM unit.

Training and Predicting a specific feature by setting PriceCategory in com.isaac.stock.predict.StockPricePrediction.java as:

PriceCategory category = PriceCategory.CLOSE; // CLOSE: train and predict close price
// or
PriceCategory category = PriceCategory.OPEN; // OPEN: train and predict open price
// ...

The PriceCategory enum:

public enum PriceCategory {
    OPEN, CLOSE, LOW, HIGH, VOLUME, ALL
}

Predicting all features as:

PriceCategory category = PriceCategory.ALL; // ALL: train and predict all features

Demo Result

A Useful GitHub Repository: timestocome/Test-stock-prediction-algorithms, which contains much information, methods and sources about predict stock and market movements.