Stock Prediction

Usage

Example

python -m utils.preprocess --targets Apple --processors technical_indicators fourier_components news_features

python -m utils.train new Apple --num-days-for-predict 60 --batch-size 32 --learning-rate 0.0001 --optimizer adamax

python -m utils.train resume model_checkpoint.ckpt

python -m utils.test lightning_logs/version_51/checkpoints/epoch=916-step=20174.ckpt

tensorboard --logdir lightning_logs

Requirements

python >= 3.10