├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── examples └── reproduce_experiment.py ├── figures └── taa.png ├── requirements.txt ├── script ├── huggingface_imbalanced.sh └── huggingface_lowresource.sh ├── setup.py └── taa ├── __init__.py ├── archive.py ├── augmentation.py ├── common.py ├── confs ├── bert_custom_data_example.yaml ├── bert_imdb_example.yaml └── bert_sst2_example.yaml ├── custom_dataset.py ├── data.py ├── data ├── .gitnore ├── custom_data_test.csv └── custom_data_train.csv ├── models └── tfidf │ └── README.md ├── search.py ├── search_and_augment.py ├── search_augment_train.py ├── text_networks ├── BERT.py └── __init__.py ├── train.py ├── transforms.py └── utils ├── __init__.py ├── analysis_results.py ├── distinct_n.py ├── get_data.py ├── metrics.py ├── raw_data_utils.py └── train_tfidf.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/README.md -------------------------------------------------------------------------------- /examples/reproduce_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/examples/reproduce_experiment.py -------------------------------------------------------------------------------- /figures/taa.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/figures/taa.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/huggingface_imbalanced.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/script/huggingface_imbalanced.sh -------------------------------------------------------------------------------- /script/huggingface_lowresource.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/script/huggingface_lowresource.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/setup.py -------------------------------------------------------------------------------- /taa/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /taa/archive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/archive.py -------------------------------------------------------------------------------- /taa/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/augmentation.py -------------------------------------------------------------------------------- /taa/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/common.py -------------------------------------------------------------------------------- /taa/confs/bert_custom_data_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/confs/bert_custom_data_example.yaml -------------------------------------------------------------------------------- /taa/confs/bert_imdb_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/confs/bert_imdb_example.yaml -------------------------------------------------------------------------------- /taa/confs/bert_sst2_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/confs/bert_sst2_example.yaml -------------------------------------------------------------------------------- /taa/custom_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/custom_dataset.py -------------------------------------------------------------------------------- /taa/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/data.py -------------------------------------------------------------------------------- /taa/data/.gitnore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore -------------------------------------------------------------------------------- /taa/data/custom_data_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/data/custom_data_test.csv -------------------------------------------------------------------------------- /taa/data/custom_data_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/data/custom_data_train.csv -------------------------------------------------------------------------------- /taa/models/tfidf/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/models/tfidf/README.md -------------------------------------------------------------------------------- /taa/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/search.py -------------------------------------------------------------------------------- /taa/search_and_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/search_and_augment.py -------------------------------------------------------------------------------- /taa/search_augment_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/search_augment_train.py -------------------------------------------------------------------------------- /taa/text_networks/BERT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/text_networks/BERT.py -------------------------------------------------------------------------------- /taa/text_networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/text_networks/__init__.py -------------------------------------------------------------------------------- /taa/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/train.py -------------------------------------------------------------------------------- /taa/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/transforms.py -------------------------------------------------------------------------------- /taa/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /taa/utils/analysis_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/utils/analysis_results.py -------------------------------------------------------------------------------- /taa/utils/distinct_n.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/utils/distinct_n.py -------------------------------------------------------------------------------- /taa/utils/get_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/utils/get_data.py -------------------------------------------------------------------------------- /taa/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/utils/metrics.py -------------------------------------------------------------------------------- /taa/utils/raw_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/utils/raw_data_utils.py -------------------------------------------------------------------------------- /taa/utils/train_tfidf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lancopku/text-autoaugment/HEAD/taa/utils/train_tfidf.py --------------------------------------------------------------------------------