/UCHRNN

An official implementation for the paper "Unsupervised Chunking with Hierarchical Recurrent Neural Networks"

Primary LanguagePythonApache License 2.0Apache-2.0

Unsupervised Chunking with Hierarchical Recurrent Neural Networks

ENV Requirement

torch
pytorch-lightning=1.6.5
transformers=4.14.1

Dataset

Download the datasets from this link

Training and Testing

For the summarization task:

bash train_summarization.sh

For the MNLI-Gen task:

bash train_paraphrasing.sh

For the translation task:

bash train_translation.sh