├── .gitignore ├── assets ├── 2022FA_appendix.pdf └── FdzRGDeUoAEtel8.jpeg ├── coop-finetune ├── README.md ├── config │ ├── optimus │ │ ├── movies_and_tv.jsonnet │ │ ├── tripadvisor.jsonnet │ │ └── yelp.jsonnet │ └── utils.libsonnet ├── evaluate.py ├── evaluations │ ├── hallucination │ │ └── evaluate_hallucination_nli.py │ └── mauve │ │ └── run_mauve.py ├── generate.py ├── setup.py └── train.py ├── coop ├── .gitignore ├── LICENSE ├── README.md ├── config │ ├── bimeanvae │ │ ├── amzn.jsonnet │ │ └── yelp.jsonnet │ ├── optimus │ │ ├── amzn.jsonnet │ │ ├── tripadvisor.jsonnet │ │ └── yelp.jsonnet │ ├── prag │ │ ├── tripadvisor.jsonnet │ │ └── yelp.jsonnet │ └── utils.libsonnet ├── coop │ ├── __init__.py │ ├── reader.py │ ├── search.py │ ├── tokenizer.py │ ├── util.py │ └── vae.py ├── evaluate.py ├── img │ └── overview.png ├── requirements.txt ├── scripts │ ├── get_summ.py │ ├── preprocess.py │ └── spm_train.py ├── setup.py └── train.py ├── diversity_utils └── evaluate_diversity.py ├── embedding_interpreter ├── .gitignore ├── huggingface_utils.py ├── modeling_bertwithoutpositionembedding.py ├── modeling_prag_decoder.py ├── readme.md ├── train_interpreter.py └── utils.py ├── evaluations ├── evaluate_hallucination_nli │ ├── evaluate_hallucination_nli.py │ └── readme.md ├── evaluate_retriever_agreement │ └── evaluate_agreement.py └── mauve │ └── run_mauve.py ├── guided_generator ├── .gitignore ├── huggingface_utils.py ├── train_generator.py └── utils.py ├── model_baselines ├── .gitignore ├── NRT │ ├── README.md │ ├── bleu.py │ ├── entailment.py │ ├── generate.py │ ├── gputils.py │ ├── hallucination_nli.py │ ├── launcher.py │ ├── main.py │ ├── module.py │ ├── nrt_predictor.py │ ├── ranking.py │ ├── rate_negations.py │ ├── rouge.py │ ├── run_evaluations.py │ └── utils.py ├── PEPLER │ ├── README.md │ ├── bleu.py │ ├── discrete.py │ ├── entailment.py │ ├── generate.py │ ├── gputils.py │ ├── hallucination_nli.py │ ├── launcher.py │ ├── main.py │ ├── mf │ │ └── yelp_1 │ │ │ └── evaluations │ │ │ └── generated.csv.diversity │ ├── module.py │ ├── pepler_predictor.py │ ├── ranking.py │ ├── rate_negations.py │ ├── reg.py │ ├── rouge.py │ ├── run_evaluations.py │ └── utils.py ├── PETER │ ├── README.md │ ├── bleu.py │ ├── entailment.py │ ├── hallucination_nli.py │ ├── main.py │ ├── module.py │ ├── peter_predictor.py │ ├── ranking.py │ ├── rate_negations.py │ ├── review_history.py │ ├── rouge.py │ ├── run_evaluations.py │ ├── streamlit_demo.py │ └── utils.py ├── PETER_F │ ├── README.md │ ├── bleu.py │ ├── entailment.py │ ├── generate.py │ ├── gputils.py │ ├── hallucination_nli.py │ ├── launcher.py │ ├── main.py │ ├── module.py │ ├── peter_predictor.py │ ├── ranking.py │ ├── rate_negations.py │ ├── rate_negations_from_generated.py │ ├── rouge.py │ ├── run_evaluations.py │ ├── streamlit_demo.py │ └── utils.py ├── att2seq │ ├── README.md │ ├── __init__.py │ ├── att2seq_predictor.py │ ├── bleu.py │ ├── entailment.py │ ├── generate.py │ ├── gputils.py │ ├── hallucination_nli.py │ ├── launcher.py │ ├── main.py │ ├── module.py │ ├── ranking.py │ ├── rate_negations.py │ ├── rouge.py │ ├── run_evaluations.py │ ├── run_test.py │ └── utils.py └── diversity_utils │ └── evaluate_diversity.py ├── noisy_sum ├── .gitignore ├── evaluations │ ├── hallucination │ │ └── evaluate_hallucination_nli.py │ └── mauve │ │ └── run_mauve.py ├── generate.py ├── generate_results.py ├── prepare_data.py └── train_summarizer.py ├── personalized_retriever ├── __init__.py ├── generate_review.py ├── huggingface_utils.py ├── modeling_bertwithoutpositionembedding.py ├── modeling_personalized_retriever.py ├── modeling_personalized_retriever_buildingblocks.py ├── readme.md ├── retrieval_pipeline.py ├── topic_extraction_pipeline.py ├── train_retriever.py └── utils.py ├── prag_generation ├── generate_review.py └── readme.md └── readme.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/.gitignore -------------------------------------------------------------------------------- /assets/2022FA_appendix.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/assets/2022FA_appendix.pdf -------------------------------------------------------------------------------- /assets/FdzRGDeUoAEtel8.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/assets/FdzRGDeUoAEtel8.jpeg -------------------------------------------------------------------------------- /coop-finetune/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/README.md -------------------------------------------------------------------------------- /coop-finetune/config/optimus/movies_and_tv.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/config/optimus/movies_and_tv.jsonnet -------------------------------------------------------------------------------- /coop-finetune/config/optimus/tripadvisor.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/config/optimus/tripadvisor.jsonnet -------------------------------------------------------------------------------- /coop-finetune/config/optimus/yelp.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/config/optimus/yelp.jsonnet -------------------------------------------------------------------------------- /coop-finetune/config/utils.libsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/config/utils.libsonnet -------------------------------------------------------------------------------- /coop-finetune/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/evaluate.py -------------------------------------------------------------------------------- /coop-finetune/evaluations/hallucination/evaluate_hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/evaluations/hallucination/evaluate_hallucination_nli.py -------------------------------------------------------------------------------- /coop-finetune/evaluations/mauve/run_mauve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/evaluations/mauve/run_mauve.py -------------------------------------------------------------------------------- /coop-finetune/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/generate.py -------------------------------------------------------------------------------- /coop-finetune/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/setup.py -------------------------------------------------------------------------------- /coop-finetune/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop-finetune/train.py -------------------------------------------------------------------------------- /coop/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/.gitignore -------------------------------------------------------------------------------- /coop/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/LICENSE -------------------------------------------------------------------------------- /coop/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/README.md -------------------------------------------------------------------------------- /coop/config/bimeanvae/amzn.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/bimeanvae/amzn.jsonnet -------------------------------------------------------------------------------- /coop/config/bimeanvae/yelp.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/bimeanvae/yelp.jsonnet -------------------------------------------------------------------------------- /coop/config/optimus/amzn.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/optimus/amzn.jsonnet -------------------------------------------------------------------------------- /coop/config/optimus/tripadvisor.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/optimus/tripadvisor.jsonnet -------------------------------------------------------------------------------- /coop/config/optimus/yelp.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/optimus/yelp.jsonnet -------------------------------------------------------------------------------- /coop/config/prag/tripadvisor.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/prag/tripadvisor.jsonnet -------------------------------------------------------------------------------- /coop/config/prag/yelp.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/prag/yelp.jsonnet -------------------------------------------------------------------------------- /coop/config/utils.libsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/config/utils.libsonnet -------------------------------------------------------------------------------- /coop/coop/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/coop/__init__.py -------------------------------------------------------------------------------- /coop/coop/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/coop/reader.py -------------------------------------------------------------------------------- /coop/coop/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/coop/search.py -------------------------------------------------------------------------------- /coop/coop/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/coop/tokenizer.py -------------------------------------------------------------------------------- /coop/coop/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/coop/util.py -------------------------------------------------------------------------------- /coop/coop/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/coop/vae.py -------------------------------------------------------------------------------- /coop/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/evaluate.py -------------------------------------------------------------------------------- /coop/img/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/img/overview.png -------------------------------------------------------------------------------- /coop/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/requirements.txt -------------------------------------------------------------------------------- /coop/scripts/get_summ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/scripts/get_summ.py -------------------------------------------------------------------------------- /coop/scripts/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/scripts/preprocess.py -------------------------------------------------------------------------------- /coop/scripts/spm_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/scripts/spm_train.py -------------------------------------------------------------------------------- /coop/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/setup.py -------------------------------------------------------------------------------- /coop/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/coop/train.py -------------------------------------------------------------------------------- /diversity_utils/evaluate_diversity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/diversity_utils/evaluate_diversity.py -------------------------------------------------------------------------------- /embedding_interpreter/.gitignore: -------------------------------------------------------------------------------- 1 | Amazon/ 2 | TripAdvisor/ 3 | Yelp/ -------------------------------------------------------------------------------- /embedding_interpreter/huggingface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/embedding_interpreter/huggingface_utils.py -------------------------------------------------------------------------------- /embedding_interpreter/modeling_bertwithoutpositionembedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/embedding_interpreter/modeling_bertwithoutpositionembedding.py -------------------------------------------------------------------------------- /embedding_interpreter/modeling_prag_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/embedding_interpreter/modeling_prag_decoder.py -------------------------------------------------------------------------------- /embedding_interpreter/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/embedding_interpreter/readme.md -------------------------------------------------------------------------------- /embedding_interpreter/train_interpreter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/embedding_interpreter/train_interpreter.py -------------------------------------------------------------------------------- /embedding_interpreter/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/embedding_interpreter/utils.py -------------------------------------------------------------------------------- /evaluations/evaluate_hallucination_nli/evaluate_hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/evaluations/evaluate_hallucination_nli/evaluate_hallucination_nli.py -------------------------------------------------------------------------------- /evaluations/evaluate_hallucination_nli/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/evaluations/evaluate_hallucination_nli/readme.md -------------------------------------------------------------------------------- /evaluations/evaluate_retriever_agreement/evaluate_agreement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/evaluations/evaluate_retriever_agreement/evaluate_agreement.py -------------------------------------------------------------------------------- /evaluations/mauve/run_mauve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/evaluations/mauve/run_mauve.py -------------------------------------------------------------------------------- /guided_generator/.gitignore: -------------------------------------------------------------------------------- 1 | **annotated_data/ -------------------------------------------------------------------------------- /guided_generator/huggingface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/guided_generator/huggingface_utils.py -------------------------------------------------------------------------------- /guided_generator/train_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/guided_generator/train_generator.py -------------------------------------------------------------------------------- /guided_generator/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/guided_generator/utils.py -------------------------------------------------------------------------------- /model_baselines/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/.gitignore -------------------------------------------------------------------------------- /model_baselines/NRT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/README.md -------------------------------------------------------------------------------- /model_baselines/NRT/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/bleu.py -------------------------------------------------------------------------------- /model_baselines/NRT/entailment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/entailment.py -------------------------------------------------------------------------------- /model_baselines/NRT/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/generate.py -------------------------------------------------------------------------------- /model_baselines/NRT/gputils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/gputils.py -------------------------------------------------------------------------------- /model_baselines/NRT/hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/hallucination_nli.py -------------------------------------------------------------------------------- /model_baselines/NRT/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/launcher.py -------------------------------------------------------------------------------- /model_baselines/NRT/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/main.py -------------------------------------------------------------------------------- /model_baselines/NRT/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/module.py -------------------------------------------------------------------------------- /model_baselines/NRT/nrt_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/nrt_predictor.py -------------------------------------------------------------------------------- /model_baselines/NRT/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/ranking.py -------------------------------------------------------------------------------- /model_baselines/NRT/rate_negations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/rate_negations.py -------------------------------------------------------------------------------- /model_baselines/NRT/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/rouge.py -------------------------------------------------------------------------------- /model_baselines/NRT/run_evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/run_evaluations.py -------------------------------------------------------------------------------- /model_baselines/NRT/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/NRT/utils.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/README.md -------------------------------------------------------------------------------- /model_baselines/PEPLER/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/bleu.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/discrete.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/entailment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/entailment.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/generate.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/gputils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/gputils.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/hallucination_nli.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/launcher.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/main.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/mf/yelp_1/evaluations/generated.csv.diversity: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/mf/yelp_1/evaluations/generated.csv.diversity -------------------------------------------------------------------------------- /model_baselines/PEPLER/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/module.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/pepler_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/pepler_predictor.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/ranking.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/rate_negations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/rate_negations.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/reg.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/rouge.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/run_evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/run_evaluations.py -------------------------------------------------------------------------------- /model_baselines/PEPLER/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PEPLER/utils.py -------------------------------------------------------------------------------- /model_baselines/PETER/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/README.md -------------------------------------------------------------------------------- /model_baselines/PETER/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/bleu.py -------------------------------------------------------------------------------- /model_baselines/PETER/entailment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/entailment.py -------------------------------------------------------------------------------- /model_baselines/PETER/hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/hallucination_nli.py -------------------------------------------------------------------------------- /model_baselines/PETER/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/main.py -------------------------------------------------------------------------------- /model_baselines/PETER/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/module.py -------------------------------------------------------------------------------- /model_baselines/PETER/peter_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/peter_predictor.py -------------------------------------------------------------------------------- /model_baselines/PETER/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/ranking.py -------------------------------------------------------------------------------- /model_baselines/PETER/rate_negations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/rate_negations.py -------------------------------------------------------------------------------- /model_baselines/PETER/review_history.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/review_history.py -------------------------------------------------------------------------------- /model_baselines/PETER/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/rouge.py -------------------------------------------------------------------------------- /model_baselines/PETER/run_evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/run_evaluations.py -------------------------------------------------------------------------------- /model_baselines/PETER/streamlit_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/streamlit_demo.py -------------------------------------------------------------------------------- /model_baselines/PETER/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER/utils.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/README.md -------------------------------------------------------------------------------- /model_baselines/PETER_F/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/bleu.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/entailment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/entailment.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/generate.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/gputils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/gputils.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/hallucination_nli.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/launcher.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/main.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/module.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/peter_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/peter_predictor.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/ranking.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/rate_negations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/rate_negations.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/rate_negations_from_generated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/rate_negations_from_generated.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/rouge.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/run_evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/run_evaluations.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/streamlit_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/streamlit_demo.py -------------------------------------------------------------------------------- /model_baselines/PETER_F/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/PETER_F/utils.py -------------------------------------------------------------------------------- /model_baselines/att2seq/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/README.md -------------------------------------------------------------------------------- /model_baselines/att2seq/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model_baselines/att2seq/att2seq_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/att2seq_predictor.py -------------------------------------------------------------------------------- /model_baselines/att2seq/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/bleu.py -------------------------------------------------------------------------------- /model_baselines/att2seq/entailment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/entailment.py -------------------------------------------------------------------------------- /model_baselines/att2seq/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/generate.py -------------------------------------------------------------------------------- /model_baselines/att2seq/gputils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/gputils.py -------------------------------------------------------------------------------- /model_baselines/att2seq/hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/hallucination_nli.py -------------------------------------------------------------------------------- /model_baselines/att2seq/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/launcher.py -------------------------------------------------------------------------------- /model_baselines/att2seq/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/main.py -------------------------------------------------------------------------------- /model_baselines/att2seq/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/module.py -------------------------------------------------------------------------------- /model_baselines/att2seq/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/ranking.py -------------------------------------------------------------------------------- /model_baselines/att2seq/rate_negations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/rate_negations.py -------------------------------------------------------------------------------- /model_baselines/att2seq/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/rouge.py -------------------------------------------------------------------------------- /model_baselines/att2seq/run_evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/run_evaluations.py -------------------------------------------------------------------------------- /model_baselines/att2seq/run_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/run_test.py -------------------------------------------------------------------------------- /model_baselines/att2seq/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/att2seq/utils.py -------------------------------------------------------------------------------- /model_baselines/diversity_utils/evaluate_diversity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/model_baselines/diversity_utils/evaluate_diversity.py -------------------------------------------------------------------------------- /noisy_sum/.gitignore: -------------------------------------------------------------------------------- 1 | **noisy_summarization_data/ -------------------------------------------------------------------------------- /noisy_sum/evaluations/hallucination/evaluate_hallucination_nli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/noisy_sum/evaluations/hallucination/evaluate_hallucination_nli.py -------------------------------------------------------------------------------- /noisy_sum/evaluations/mauve/run_mauve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/noisy_sum/evaluations/mauve/run_mauve.py -------------------------------------------------------------------------------- /noisy_sum/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/noisy_sum/generate.py -------------------------------------------------------------------------------- /noisy_sum/generate_results.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /noisy_sum/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/noisy_sum/prepare_data.py -------------------------------------------------------------------------------- /noisy_sum/train_summarizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/noisy_sum/train_summarizer.py -------------------------------------------------------------------------------- /personalized_retriever/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /personalized_retriever/generate_review.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/generate_review.py -------------------------------------------------------------------------------- /personalized_retriever/huggingface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/huggingface_utils.py -------------------------------------------------------------------------------- /personalized_retriever/modeling_bertwithoutpositionembedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/modeling_bertwithoutpositionembedding.py -------------------------------------------------------------------------------- /personalized_retriever/modeling_personalized_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/modeling_personalized_retriever.py -------------------------------------------------------------------------------- /personalized_retriever/modeling_personalized_retriever_buildingblocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/modeling_personalized_retriever_buildingblocks.py -------------------------------------------------------------------------------- /personalized_retriever/readme.md: -------------------------------------------------------------------------------- 1 | entry point: train_retriever.py 2 | -------------------------------------------------------------------------------- /personalized_retriever/retrieval_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/retrieval_pipeline.py -------------------------------------------------------------------------------- /personalized_retriever/topic_extraction_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/topic_extraction_pipeline.py -------------------------------------------------------------------------------- /personalized_retriever/train_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/train_retriever.py -------------------------------------------------------------------------------- /personalized_retriever/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/personalized_retriever/utils.py -------------------------------------------------------------------------------- /prag_generation/generate_review.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/prag_generation/generate_review.py -------------------------------------------------------------------------------- /prag_generation/readme.md: -------------------------------------------------------------------------------- 1 | ## Place for handling text generation, etc -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhouhanxie/PRAG/HEAD/readme.md --------------------------------------------------------------------------------