├── README.md ├── requirements.txt └── src ├── bart_aug ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── denoised_dataset.cpython-37.pyc │ └── masking_task.cpython-37.pyc ├── denoised_dataset.py └── masking_task.py ├── bert_aug ├── __init__.py ├── __pycache__ │ ├── bert_model.cpython-37.pyc │ ├── bert_model.cpython-38.pyc │ ├── data_processors.cpython-37.pyc │ └── data_processors.cpython-38.pyc ├── backtranslation.py ├── bert_classifier.py ├── bert_model.py ├── cbert.py ├── cgpt2.py ├── cmodbert.py ├── cmodbertp.py ├── data_processors.py └── eda.py ├── scripts ├── bart_snips_lower.sh ├── bart_stsa_lower.sh ├── bart_trec_lower.sh ├── bert_snips_lower.sh ├── bert_stsa_lower.sh └── bert_trec_lower.sh └── utils ├── __init__.py ├── bpe_encoder.py ├── convert_num_to_text_labels.py ├── create_fsl_dataset.py ├── download_and_prepare_datasets.sh └── gpt2_bpe ├── dict.txt ├── encoder.json └── vocab.bpe /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | #pytorch==1.6 2 | fairseq==0.9 3 | transformers==2.9 -------------------------------------------------------------------------------- /src/bart_aug/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bart_aug/__init__.py -------------------------------------------------------------------------------- /src/bart_aug/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bart_aug/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /src/bart_aug/__pycache__/denoised_dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bart_aug/__pycache__/denoised_dataset.cpython-37.pyc -------------------------------------------------------------------------------- /src/bart_aug/__pycache__/masking_task.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bart_aug/__pycache__/masking_task.cpython-37.pyc -------------------------------------------------------------------------------- /src/bart_aug/denoised_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bart_aug/denoised_dataset.py -------------------------------------------------------------------------------- /src/bart_aug/masking_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bart_aug/masking_task.py -------------------------------------------------------------------------------- /src/bert_aug/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/bert_aug/__pycache__/bert_model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/__pycache__/bert_model.cpython-37.pyc -------------------------------------------------------------------------------- /src/bert_aug/__pycache__/bert_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/__pycache__/bert_model.cpython-38.pyc -------------------------------------------------------------------------------- /src/bert_aug/__pycache__/data_processors.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/__pycache__/data_processors.cpython-37.pyc -------------------------------------------------------------------------------- /src/bert_aug/__pycache__/data_processors.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/__pycache__/data_processors.cpython-38.pyc -------------------------------------------------------------------------------- /src/bert_aug/backtranslation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/backtranslation.py -------------------------------------------------------------------------------- /src/bert_aug/bert_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/bert_classifier.py -------------------------------------------------------------------------------- /src/bert_aug/bert_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/bert_model.py -------------------------------------------------------------------------------- /src/bert_aug/cbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/cbert.py -------------------------------------------------------------------------------- /src/bert_aug/cgpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/cgpt2.py -------------------------------------------------------------------------------- /src/bert_aug/cmodbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/cmodbert.py -------------------------------------------------------------------------------- /src/bert_aug/cmodbertp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/cmodbertp.py -------------------------------------------------------------------------------- /src/bert_aug/data_processors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/data_processors.py -------------------------------------------------------------------------------- /src/bert_aug/eda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/bert_aug/eda.py -------------------------------------------------------------------------------- /src/scripts/bart_snips_lower.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/scripts/bart_snips_lower.sh -------------------------------------------------------------------------------- /src/scripts/bart_stsa_lower.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/scripts/bart_stsa_lower.sh -------------------------------------------------------------------------------- /src/scripts/bart_trec_lower.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/scripts/bart_trec_lower.sh -------------------------------------------------------------------------------- /src/scripts/bert_snips_lower.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/scripts/bert_snips_lower.sh -------------------------------------------------------------------------------- /src/scripts/bert_stsa_lower.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/scripts/bert_stsa_lower.sh -------------------------------------------------------------------------------- /src/scripts/bert_trec_lower.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/scripts/bert_trec_lower.sh -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/bpe_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/utils/bpe_encoder.py -------------------------------------------------------------------------------- /src/utils/convert_num_to_text_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/utils/convert_num_to_text_labels.py -------------------------------------------------------------------------------- /src/utils/create_fsl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/utils/create_fsl_dataset.py -------------------------------------------------------------------------------- /src/utils/download_and_prepare_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/utils/download_and_prepare_datasets.sh -------------------------------------------------------------------------------- /src/utils/gpt2_bpe/dict.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/utils/gpt2_bpe/dict.txt -------------------------------------------------------------------------------- /src/utils/gpt2_bpe/encoder.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/utils/gpt2_bpe/encoder.json -------------------------------------------------------------------------------- /src/utils/gpt2_bpe/vocab.bpe: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caskcsg/TextSmoothing/HEAD/src/utils/gpt2_bpe/vocab.bpe --------------------------------------------------------------------------------