├── 000000.py ├── README.md ├── __init__.py ├── __pycache__ └── parameter_config.cpython-311.pyc ├── app.py ├── config └── config.json ├── data ├── medical_train.txt └── medical_valid.txt ├── data_preprocess ├── __init__.py ├── dataloader.py ├── dataset.py └── preprocess.py ├── flask_predict.py ├── functions_tools.py ├── gpt2 ├── README.md ├── generation_config.json ├── merges.txt ├── tokenizer.json └── vocab.json ├── interact.py ├── parameter_config.py ├── pytorch_tools.py ├── templates └── index.html ├── test.py ├── train.py └── vocab ├── vocab.txt └── vocab2.txt /000000.py: -------------------------------------------------------------------------------- 1 | import torch 2 | print(torch.version.cuda) # 输出CUDA版本 -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/__init__.py -------------------------------------------------------------------------------- /__pycache__/parameter_config.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/__pycache__/parameter_config.cpython-311.pyc -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/app.py -------------------------------------------------------------------------------- /config/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/config/config.json -------------------------------------------------------------------------------- /data/medical_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/data/medical_train.txt -------------------------------------------------------------------------------- /data/medical_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/data/medical_valid.txt -------------------------------------------------------------------------------- /data_preprocess/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/data_preprocess/__init__.py -------------------------------------------------------------------------------- /data_preprocess/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/data_preprocess/dataloader.py -------------------------------------------------------------------------------- /data_preprocess/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/data_preprocess/dataset.py -------------------------------------------------------------------------------- /data_preprocess/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/data_preprocess/preprocess.py -------------------------------------------------------------------------------- /flask_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/flask_predict.py -------------------------------------------------------------------------------- /functions_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/functions_tools.py -------------------------------------------------------------------------------- /gpt2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/gpt2/README.md -------------------------------------------------------------------------------- /gpt2/generation_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/gpt2/generation_config.json -------------------------------------------------------------------------------- /gpt2/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/gpt2/merges.txt -------------------------------------------------------------------------------- /gpt2/tokenizer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/gpt2/tokenizer.json -------------------------------------------------------------------------------- /gpt2/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/gpt2/vocab.json -------------------------------------------------------------------------------- /interact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/interact.py -------------------------------------------------------------------------------- /parameter_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/parameter_config.py -------------------------------------------------------------------------------- /pytorch_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/pytorch_tools.py -------------------------------------------------------------------------------- /templates/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/templates/index.html -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/train.py -------------------------------------------------------------------------------- /vocab/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/vocab/vocab.txt -------------------------------------------------------------------------------- /vocab/vocab2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Haohao-end/medical-inquiry-robot-based-on-GPT2/HEAD/vocab/vocab2.txt --------------------------------------------------------------------------------