pytorch bert文本分类_听说你想用 Bert 进行文本分类

03952984fddb72f5938cb08db5f6c66f.png

最近一直在搞文本分类,别问为啥,问就是生活所迫。 前期写了几个简单模型之后,就想着试试 Bert 在文本分类上的表现,然后我去看了 pytorch-pretrained-BERT, 然后, 我去, 这写的有点随心所欲呀,改起来岂不是要吐血.

因此,我想着我自己把它做一个简要版, 删除了分布式逻辑以及半浮点数逻辑,并对整个进行了封装, 现在你要想适配一个对应的数据集,只需要两步,简直不要太简单。

后续这个仓库还会不断在Bert之上做文章,比如加一些CNN之类的,后续会出一揽子计划,静待更新。 说了这么多,仓库地址: Bert-TextClassification, 欢迎大家一起来搞事情哦。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值