The implementation of the paper "HIST: A Graph-based Framework for Stock Trend Forecasting via Mining Concept-Oriented Shared Information".

Last update: Dec 27, 2022

Related tags

Deep Learning HIST

Overview

The HIST framework for stock trend forecasting

The implementation of the paper "HIST: A Graph-based Framework for Stock Trend Forecasting via Mining Concept-Oriented Shared Information".

Environment

Install python3.7, 3.8 or 3.9.
Install the requirements in requirements.txt.

Install the Qlib and download the data:

# install Qlib from source
pip install --upgrade  cython
git clone https://github.com/microsoft/qlib.git && cd qlib
python setup.py install

# Download the stock features of Alpha360
python scripts/get_data.py qlib_data --target_dir ~/.qlib/qlib_data/cn_data --region cn --version v2

Reproduce the stock trend forecasting results

git clone https://github.com/Wentao-Xu/HIST.git
cd HIST
mkdir output

Reproduce our HIST framework

# CSI 100
python learn.py --model_name HIST --data_set csi100 --hidden_size 128 --num_layers 2 --outdir ./output/csi100_HIST

# CSI 300
python learn.py --model_name HIST --data_set csi300 --hidden_size 128 --num_layers 2 --outdir ./output/csi300_HIST

Reproduce the baselines

# MLP on CSI 100
python learn.py --model_name MLP --data_set csi100 --hidden_size 512 --num_layers 3 --outdir ./output/csi100_MLP

# MLP on CSI 300
python learn.py --model_name MLP --data_set csi300 --hidden_size 512 --num_layers 3 --outdir ./output/csi300_MLP

LSTM

# LSTM on CSI 100
python learn.py --model_name LSTM --data_set csi100 --hidden_size 128 --num_layers 2 --outdir ./output/csi100_LSTM

# LSTM on CSI 300
python learn.py --model_name LSTM --data_set csi300 --hidden_size 128 --num_layers 2 --outdir ./output/csi300_LSTM

# GRU on CSI 100
python learn.py --model_name GRU --data_set csi100 --hidden_size 128 --num_layers 2 --outdir ./output/csi100_GRU

# GRU on CSI 300
python learn.py --model_name GRU --data_set csi300 --hidden_size 64 --num_layers 2 --outdir ./output/csi300_GRU

# SFM on CSI 100
python learn.py --model_name SFM --data_set csi100 --hidden_size 64 --num_layers 2 --outdir ./output/csi100_SFM

# SFM on CSI 300
python learn.py --model_name SFM --data_set csi300 --hidden_size 128 --num_layers 2 --outdir ./output/csi300_SFM

GATs

# GATs on CSI 100
python learn.py --model_name GATs --data_set csi100 --hidden_size 128 --num_layers 2 --outdir ./output/csi100_GATs

# GATs on CSI 300
python learn.py --model_name GATs --data_set csi300 --hidden_size 64 --num_layers 2 --outdir ./output/csi300_GATs

ALSTM

# ALSTM on CSI 100
python learn.py --model_name ALSTM --data_set csi100 --hidden_size 64 --num_layers 2 --outdir ./output/csi100_ALSTM

# ALSTM on CSI 300
python learn.py --model_name ALSTM --data_set csi300 --hidden_size 128 --num_layers 2 --outdir ./output/csi300_ALSTM

Transformer

# Transformer on CSI 100
python learn.py --model_name Transformer --data_set csi100 --hidden_size 32 --num_layers 3 --outdir ./output/csi100_Transformer

# Transformer on CSI 300
python learn.py --model_name Transformer --data_set csi300 --hidden_size 32 --num_layers 3 --outdir ./output/csi300_Transformer

ALSTM+TRA

We reproduce the ALSTM+TRA with its source code.

The implementation of the paper "HIST: A Graph-based Framework for Stock Trend Forecasting via Mining Concept-Oriented Shared Information".

Related tags

Overview

The HIST framework for stock trend forecasting

Environment

Reproduce the stock trend forecasting results

Reproduce our HIST framework

Reproduce the baselines

Owner

Wentao Xu

Learning from Guided Play: A Scheduled Hierarchical Approach for Improving Exploration in Adversarial Imitation Learning Source Code

A simple image/video to Desmos graph converter run locally

Mini-hmc-jax - A simple implementation of Hamiltonian Monte Carlo in JAX

Implementations of CNNs, RNNs, GANs, etc

Preparation material for Dropbox interviews

NLP From Scratch Without Large-Scale Pretraining: A Simple and Efficient Framework

My Body is a Cage: the Role of Morphology in Graph-Based Incompatible Control

Vehicles Counting using YOLOv4 + DeepSORT + Flask + Ngrok

Puzzle-CAM: Improved localization via matching partial and full features.

Code for Pose-Controllable Talking Face Generation by Implicitly Modularized Audio-Visual Representation (CVPR 2021)

Worktory is a python library created with the single purpose of simplifying the inventory management of network automation scripts.

Pytorch implementation for Patient Knowledge Distillation for BERT Model Compression

Realtime YOLO Monster Detection With Non Maximum Supression

Teaches a student network from the knowledge obtained via training of a larger teacher network

SubOmiEmbed: Self-supervised Representation Learning of Multi-omics Data for Cancer Type Classification

The official homepage of the (outdated) COCO-Stuff 10K dataset.

Predicting future trajectories of people in cameras of novel scenarios and views.

Code to reproduce the results in "Visually Grounded Reasoning across Languages and Cultures", EMNLP 2021.

SegTransVAE: Hybrid CNN - Transformer with Regularization for medical image segmentation

Dirty Pixels: Towards End-to-End Image Processing and Perception