这个代码是文本分类的，可以把它改成实体识别的吗？改的地方多吗

[Zhang Xiaojun]