├── .idea ├── .gitignore ├── UnilmChatchitRobot.iml ├── deployment.xml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── LICENSE ├── README.md ├── configuration_unilm.py ├── data ├── dirty_words.txt ├── douban_kuakua_qa.txt └── sample.json ├── dirty_recognize.py ├── image ├── logcong.png └── tensorboard.png ├── interactive_conditional_samples.py ├── kuakua_robot_model ├── config.json └── vocab.txt ├── modeling_unilm.py ├── requirements.txt ├── run_train.py ├── run_train.sh ├── tokenization_unilm.py ├── trie.py ├── unilm_model ├── config.json └── vocab.txt └── utils_seq2seq.py /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | 2 | # Default ignored files 3 | /workspace.xml -------------------------------------------------------------------------------- /.idea/UnilmChatchitRobot.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/.idea/UnilmChatchitRobot.iml -------------------------------------------------------------------------------- /.idea/deployment.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/.idea/deployment.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/README.md -------------------------------------------------------------------------------- /configuration_unilm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/configuration_unilm.py -------------------------------------------------------------------------------- /data/dirty_words.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/data/dirty_words.txt -------------------------------------------------------------------------------- /data/douban_kuakua_qa.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/data/douban_kuakua_qa.txt -------------------------------------------------------------------------------- /data/sample.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/data/sample.json -------------------------------------------------------------------------------- /dirty_recognize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/dirty_recognize.py -------------------------------------------------------------------------------- /image/logcong.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/image/logcong.png -------------------------------------------------------------------------------- /image/tensorboard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/image/tensorboard.png -------------------------------------------------------------------------------- /interactive_conditional_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/interactive_conditional_samples.py -------------------------------------------------------------------------------- /kuakua_robot_model/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/kuakua_robot_model/config.json -------------------------------------------------------------------------------- /kuakua_robot_model/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/kuakua_robot_model/vocab.txt -------------------------------------------------------------------------------- /modeling_unilm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/modeling_unilm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | transformers == 3.0.2 2 | pytorch >= 1.4 3 | tqdm 4 | numpy 5 | -------------------------------------------------------------------------------- /run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/run_train.py -------------------------------------------------------------------------------- /run_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/run_train.sh -------------------------------------------------------------------------------- /tokenization_unilm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/tokenization_unilm.py -------------------------------------------------------------------------------- /trie.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/trie.py -------------------------------------------------------------------------------- /unilm_model/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/unilm_model/config.json -------------------------------------------------------------------------------- /unilm_model/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/unilm_model/vocab.txt -------------------------------------------------------------------------------- /utils_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liucongg/UnilmChatchitRobot/HEAD/utils_seq2seq.py --------------------------------------------------------------------------------