No description, website, or topics provided.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data
.gitignore
LICENSE
MultiClassNLLCriterion.lua
README.md
accuracy.lua
lstmv2.l_2.t_11.rnn_425.uniform.3.log
model.gru.lua
model.gruv2.lua
model.lstm.lua
model.lstmv2.lua
myrmsprop.lua
test.lua
test.py
test_samples.mat
train.lua

README.md

SBL-LSTM-Net

This is the original implementation of the deep nerual network model proposed in our paper, "From Bayesian Sparsity to Gated Recurrent Nets"[Paper].

If you find this code useful in your research, please cite:

@article{he2017bayesian,
  title={From Bayesian Sparsity to Gated Recurrent Nets},
  author={He, Hao and Xin, Bo and Wipf, David},
  journal={arXiv preprint arXiv:1706.02815},
  year={2017}
}

Requirements

  • Linux or macOS
  • Torch 7
  • Python 2 or 3
  • CPU or NVIDIA GPU + CUDA CuDNN

Get started

Train/Test