![03952984fddb72f5938cb08db5f6c66f.png](https://i-blog.csdnimg.cn/blog_migrate/3a5ce826fb022befc4721b48ad1913ec.jpeg)
最近一直在搞文本分类,别问为啥,问就是生活所迫。 前期写了几个简单模型之后,就想着试试 Bert 在文本分类上的表现,然后我去看了 pytorch-pretrained-BERT, 然后, 我去, 这写的有点随心所欲呀,改起来岂不是要吐血.
因此,我想着我自己把它做一个简要版, 删除了分布式逻辑以及半浮点数逻辑,并对整个进行了封装, 现在你要想适配一个对应的数据集,只需要两步,简直不要太简单。
后续这个仓库还会不断在Bert之上做文章,比如加一些CNN之类的,后续会出一揽子计划,静待更新。 说了这么多,仓库地址: Bert-TextClassification, 欢迎大家一起来搞事情哦。