├── GenAug_chart.png ├── README.md ├── code ├── augmentation_methods │ ├── GenAug_Random_Trio.ipynb │ ├── GenAug_Synthetic_Noise.ipynb │ ├── GenAug_WordNet_Hypernyms.ipynb │ ├── GenAug_WordNet_Hyponyms.ipynb │ └── GenAug_WordNet_Synonyms.ipynb ├── evaluation │ ├── evaluate_BPRO.py │ ├── evaluate_PPL_SLOR.py │ ├── evaluate_combined.py │ ├── evaluate_spellcheck.py │ ├── get_unigram_distribution.py │ ├── metric_utils.py │ └── sentiment │ │ ├── examples │ │ └── run_glue.py │ │ ├── finetune_classifier.sh │ │ ├── flatten_generations.py │ │ ├── prepare_output_for_classifier.py │ │ ├── prepare_yelp_classifier_data.sh │ │ ├── regenerate_generations.py │ │ ├── regenerate_gold.py │ │ ├── run_aspect_preservation_generations.sh │ │ ├── run_aspect_preservation_gold.sh │ │ └── sentiment_analysis.py ├── finetuning_and_generation │ ├── examples │ │ ├── run_generation_bounded.py │ │ └── run_language_modeling.py │ ├── finetune_generator_gpt2.sh │ └── generate_generator_gpt2.sh └── processing_and_setup │ ├── Continuation_Postprocessor.py │ ├── GenAug_Finetuning_Setup.ipynb │ ├── GenAug_SMERTI_Processing.ipynb │ └── GenAug_Yelp_Dataset_Processing.ipynb ├── data ├── SMERTI_chosen_REs.txt ├── SMERTI_train.csv ├── SMERTI_val.csv ├── stopwords.txt ├── yelp_test.txt ├── yelp_test_0.5.txt ├── yelp_train.txt ├── yelp_train_0.5.txt └── yelp_val.txt └── requirements.txt /GenAug_chart.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/GenAug_chart.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/README.md -------------------------------------------------------------------------------- /code/augmentation_methods/GenAug_Random_Trio.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/augmentation_methods/GenAug_Random_Trio.ipynb -------------------------------------------------------------------------------- /code/augmentation_methods/GenAug_Synthetic_Noise.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/augmentation_methods/GenAug_Synthetic_Noise.ipynb -------------------------------------------------------------------------------- /code/augmentation_methods/GenAug_WordNet_Hypernyms.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/augmentation_methods/GenAug_WordNet_Hypernyms.ipynb -------------------------------------------------------------------------------- /code/augmentation_methods/GenAug_WordNet_Hyponyms.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/augmentation_methods/GenAug_WordNet_Hyponyms.ipynb -------------------------------------------------------------------------------- /code/augmentation_methods/GenAug_WordNet_Synonyms.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/augmentation_methods/GenAug_WordNet_Synonyms.ipynb -------------------------------------------------------------------------------- /code/evaluation/evaluate_BPRO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/evaluate_BPRO.py -------------------------------------------------------------------------------- /code/evaluation/evaluate_PPL_SLOR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/evaluate_PPL_SLOR.py -------------------------------------------------------------------------------- /code/evaluation/evaluate_combined.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/evaluate_combined.py -------------------------------------------------------------------------------- /code/evaluation/evaluate_spellcheck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/evaluate_spellcheck.py -------------------------------------------------------------------------------- /code/evaluation/get_unigram_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/get_unigram_distribution.py -------------------------------------------------------------------------------- /code/evaluation/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/metric_utils.py -------------------------------------------------------------------------------- /code/evaluation/sentiment/examples/run_glue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/examples/run_glue.py -------------------------------------------------------------------------------- /code/evaluation/sentiment/finetune_classifier.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/finetune_classifier.sh -------------------------------------------------------------------------------- /code/evaluation/sentiment/flatten_generations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/flatten_generations.py -------------------------------------------------------------------------------- /code/evaluation/sentiment/prepare_output_for_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/prepare_output_for_classifier.py -------------------------------------------------------------------------------- /code/evaluation/sentiment/prepare_yelp_classifier_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/prepare_yelp_classifier_data.sh -------------------------------------------------------------------------------- /code/evaluation/sentiment/regenerate_generations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/regenerate_generations.py -------------------------------------------------------------------------------- /code/evaluation/sentiment/regenerate_gold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/regenerate_gold.py -------------------------------------------------------------------------------- /code/evaluation/sentiment/run_aspect_preservation_generations.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/run_aspect_preservation_generations.sh -------------------------------------------------------------------------------- /code/evaluation/sentiment/run_aspect_preservation_gold.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/run_aspect_preservation_gold.sh -------------------------------------------------------------------------------- /code/evaluation/sentiment/sentiment_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/evaluation/sentiment/sentiment_analysis.py -------------------------------------------------------------------------------- /code/finetuning_and_generation/examples/run_generation_bounded.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/finetuning_and_generation/examples/run_generation_bounded.py -------------------------------------------------------------------------------- /code/finetuning_and_generation/examples/run_language_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/finetuning_and_generation/examples/run_language_modeling.py -------------------------------------------------------------------------------- /code/finetuning_and_generation/finetune_generator_gpt2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/finetuning_and_generation/finetune_generator_gpt2.sh -------------------------------------------------------------------------------- /code/finetuning_and_generation/generate_generator_gpt2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/finetuning_and_generation/generate_generator_gpt2.sh -------------------------------------------------------------------------------- /code/processing_and_setup/Continuation_Postprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/processing_and_setup/Continuation_Postprocessor.py -------------------------------------------------------------------------------- /code/processing_and_setup/GenAug_Finetuning_Setup.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/processing_and_setup/GenAug_Finetuning_Setup.ipynb -------------------------------------------------------------------------------- /code/processing_and_setup/GenAug_SMERTI_Processing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/processing_and_setup/GenAug_SMERTI_Processing.ipynb -------------------------------------------------------------------------------- /code/processing_and_setup/GenAug_Yelp_Dataset_Processing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/code/processing_and_setup/GenAug_Yelp_Dataset_Processing.ipynb -------------------------------------------------------------------------------- /data/SMERTI_chosen_REs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/SMERTI_chosen_REs.txt -------------------------------------------------------------------------------- /data/SMERTI_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/SMERTI_train.csv -------------------------------------------------------------------------------- /data/SMERTI_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/SMERTI_val.csv -------------------------------------------------------------------------------- /data/stopwords.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/stopwords.txt -------------------------------------------------------------------------------- /data/yelp_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/yelp_test.txt -------------------------------------------------------------------------------- /data/yelp_test_0.5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/yelp_test_0.5.txt -------------------------------------------------------------------------------- /data/yelp_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/yelp_train.txt -------------------------------------------------------------------------------- /data/yelp_train_0.5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/yelp_train_0.5.txt -------------------------------------------------------------------------------- /data/yelp_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/data/yelp_val.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/styfeng/GenAug/HEAD/requirements.txt --------------------------------------------------------------------------------