https://github.com/shibing624/pytextclassifier/blob/master/examples/bert_classification_zh_demo.py