RNN.md
December 27, 2022 ยท View on GitHub
RNN
You can train a RNN encoder-decoder with attention from scratch with this model. Three models are available:
- RNN
- GRU
- LSTM
You can choose them through model=RNN,model=GRU,model=LSTM. Meanwhile, you can check or modify the default parameters of the model in textbox/property/model/rnn.yaml(gru.yaml)(lstm.yaml)
Example usage:
python run_textbox.py \
--model=RNN \
--dataset=samsum