like0403 / attentioned-dual-stage-stock-prediction Goto Github PK

This is an implementation of paper "A Dual-Stage Attention-Based Recurrent Neural Network for Time Series Prediction". I only did a test to predict the price of AAPL.US by its historical data as well as the price of its opponent MSFT.US.

Downloaded from NASDAQ 100 STOCK DATA.

-e, --epoch - the number of epochs

-b, --batch - the batch size

-s, --split - the split ratio of train and test set

-i, --interval - save models every interval epochs

-l, --lrate - learning rate of optimizor

-t, —test - test phase

-m, —model - if in test phase, the models name(if model name is "encoder50" and decoder50", inptut 50)

Traing 500 epochs, with batch-size 128, save models every 100 epochs.

Python3 trainer -e 500 -b 128 -i 100

Test data use model "encoder50" and "decoder50"

Python3 trainer -t -m 50