Skip to content

A Dual-Stage Attention-Based Recurrent Neural Network for Time Series Prediction

Notifications You must be signed in to change notification settings

Chromatius/da-rnn

Repository files navigation

da-rnn

A Dual-Stage Attention-Based Recurrent Neural Network for Time Series Prediction

Implements the algorithm described in https://arxiv.org/pdf/1704.02971.pdf in Pytorch. The implementation also includes a Dataset class with additional utility for handling messy data sets. The implementation also includes an implementation of Adam with weight decay and Cosine annealing to increase training speed.

About

A Dual-Stage Attention-Based Recurrent Neural Network for Time Series Prediction

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published