• 优雅の使用transformer系列之text-classification
• 优雅の使用transformer系列之gpt2-chitchat

这个repo会持续放基于transformer和pytorch-lightning的一些功能实现，也欢迎大家contribute.

• python >= 3.6
• pytorch

pip install -r requirements.txt

下载依赖package

python run_classification_pl.py

关于bert预训练模型，可以通过指定'--model_name_or_path=YOUR DOWNLOAD BERT MODEL PATH'来避免去外网下载，这里的训练数据来自rasa-nlu-benchmark

python run_classification_pl.py --do_train=False --do_predict=True

http://localhost:5000/parse post请求，请求参数例如：

["糖醋排骨怎么做啊？"]

当然也可以使用postman去请求调用

python run_ner_pl.py

关于bert预训练模型，可以通过指定'--model_name_or_path=YOUR DOWNLOAD BERT MODEL PATH'来避免去外网下载

python run_ner_pl.py --do_train=False --do_predict=True

具体的可以参考优雅の使用transformer系列之gpt2-chitchat这篇文章