├── .gitignore ├── LICENSE ├── fig └── model.png ├── generator ├── bart_generation.py ├── chunk_dataset.py ├── fid_model.py ├── readme.md ├── run_model_test.py └── run_model_train.py ├── readme.md └── retriever ├── eval.py ├── index.py ├── index.sh ├── models ├── __init__.py ├── loss_func.py └── regen_retriever.py ├── readme.md ├── train.py ├── train.sh └── utils ├── __init__.py ├── args.py ├── dataset.py ├── retrieval_utils.py └── sys_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/LICENSE -------------------------------------------------------------------------------- /fig/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/fig/model.png -------------------------------------------------------------------------------- /generator/bart_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/generator/bart_generation.py -------------------------------------------------------------------------------- /generator/chunk_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/generator/chunk_dataset.py -------------------------------------------------------------------------------- /generator/fid_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/generator/fid_model.py -------------------------------------------------------------------------------- /generator/readme.md: -------------------------------------------------------------------------------- 1 | This is the code for retriever. 2 | -------------------------------------------------------------------------------- /generator/run_model_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/generator/run_model_test.py -------------------------------------------------------------------------------- /generator/run_model_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/generator/run_model_train.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/readme.md -------------------------------------------------------------------------------- /retriever/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/eval.py -------------------------------------------------------------------------------- /retriever/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/index.py -------------------------------------------------------------------------------- /retriever/index.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/index.sh -------------------------------------------------------------------------------- /retriever/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/models/__init__.py -------------------------------------------------------------------------------- /retriever/models/loss_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/models/loss_func.py -------------------------------------------------------------------------------- /retriever/models/regen_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/models/regen_retriever.py -------------------------------------------------------------------------------- /retriever/readme.md: -------------------------------------------------------------------------------- 1 | for retriever 2 | -------------------------------------------------------------------------------- /retriever/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/train.py -------------------------------------------------------------------------------- /retriever/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/train.sh -------------------------------------------------------------------------------- /retriever/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/utils/__init__.py -------------------------------------------------------------------------------- /retriever/utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/utils/args.py -------------------------------------------------------------------------------- /retriever/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/utils/dataset.py -------------------------------------------------------------------------------- /retriever/utils/retrieval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/utils/retrieval_utils.py -------------------------------------------------------------------------------- /retriever/utils/sys_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qhjqhj00/WebBrain/HEAD/retriever/utils/sys_utils.py --------------------------------------------------------------------------------