├── README.md ├── config ├── config_ptb.py ├── config_short_yelp.py ├── config_snli.py ├── config_yahoo.py └── config_yahoo_label.py ├── data ├── __init__.py └── text_data.py ├── exp_utils.py ├── lm.py ├── modules ├── __init__.py ├── decoders │ ├── __init__.py │ ├── dec_lstm.py │ ├── decoder.py │ ├── decoder_help.py │ └── decoder_helper.py ├── discriminators │ ├── __init__.py │ └── discriminator_linear.py ├── encoders │ ├── __init__.py │ ├── enc_lstm.py │ ├── encoder.py │ └── gaussian_encoder.py ├── lm │ ├── __init__.py │ └── lm_lstm.py ├── utils.py └── vae.py ├── prepare_data.py ├── scripts ├── multi-bleu.perl ├── sampling_training_labels.py └── unsupervised_cluster.py ├── text_anneal_fb.py ├── text_beta.py ├── text_get_mean.py ├── text_ss_ft.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/README.md -------------------------------------------------------------------------------- /config/config_ptb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/config/config_ptb.py -------------------------------------------------------------------------------- /config/config_short_yelp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/config/config_short_yelp.py -------------------------------------------------------------------------------- /config/config_snli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/config/config_snli.py -------------------------------------------------------------------------------- /config/config_yahoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/config/config_yahoo.py -------------------------------------------------------------------------------- /config/config_yahoo_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/config/config_yahoo_label.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | from .text_data import * -------------------------------------------------------------------------------- /data/text_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/data/text_data.py -------------------------------------------------------------------------------- /exp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/exp_utils.py -------------------------------------------------------------------------------- /lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/lm.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/__init__.py -------------------------------------------------------------------------------- /modules/decoders/__init__.py: -------------------------------------------------------------------------------- 1 | from .dec_lstm import * 2 | -------------------------------------------------------------------------------- /modules/decoders/dec_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/decoders/dec_lstm.py -------------------------------------------------------------------------------- /modules/decoders/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/decoders/decoder.py -------------------------------------------------------------------------------- /modules/decoders/decoder_help.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/decoders/decoder_help.py -------------------------------------------------------------------------------- /modules/decoders/decoder_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/decoders/decoder_helper.py -------------------------------------------------------------------------------- /modules/discriminators/__init__.py: -------------------------------------------------------------------------------- 1 | from .discriminator_linear import * -------------------------------------------------------------------------------- /modules/discriminators/discriminator_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/discriminators/discriminator_linear.py -------------------------------------------------------------------------------- /modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | from .enc_lstm import * 2 | -------------------------------------------------------------------------------- /modules/encoders/enc_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/encoders/enc_lstm.py -------------------------------------------------------------------------------- /modules/encoders/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/encoders/encoder.py -------------------------------------------------------------------------------- /modules/encoders/gaussian_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/encoders/gaussian_encoder.py -------------------------------------------------------------------------------- /modules/lm/__init__.py: -------------------------------------------------------------------------------- 1 | from .lm_lstm import * -------------------------------------------------------------------------------- /modules/lm/lm_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/lm/lm_lstm.py -------------------------------------------------------------------------------- /modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/utils.py -------------------------------------------------------------------------------- /modules/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/modules/vae.py -------------------------------------------------------------------------------- /prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/prepare_data.py -------------------------------------------------------------------------------- /scripts/multi-bleu.perl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/scripts/multi-bleu.perl -------------------------------------------------------------------------------- /scripts/sampling_training_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/scripts/sampling_training_labels.py -------------------------------------------------------------------------------- /scripts/unsupervised_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/scripts/unsupervised_cluster.py -------------------------------------------------------------------------------- /text_anneal_fb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/text_anneal_fb.py -------------------------------------------------------------------------------- /text_beta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/text_beta.py -------------------------------------------------------------------------------- /text_get_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/text_get_mean.py -------------------------------------------------------------------------------- /text_ss_ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/text_ss_ft.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bohanli/vae-pretraining-encoder/HEAD/utils.py --------------------------------------------------------------------------------