Tensorflow CNN文本分类项目
完整的 CNN 文本分类项目,用的就是咱们熟悉的 Tensorflow,整套流程跑通,包括数据预、模型构建、训练和评估,结构清晰,逻辑顺。
代码量不大,但功能还挺全。比如用 Conv1D
搭了个多通道卷积,配上 Embedding
和 GlobalMaxPooling1D
,跑短文本还挺快的,准确率也不错。
想练练手的,可以从这项目下手,改起来也方便。尤其你要是做 NLP,这个工程用来调试各种超参合适。
相关资料也都整理好了,像Tensorflow 文本分类实战、CNN 文本分类模型训练这些,都是配套的,跟着做效率高。
如果你对 BERT 或 Transformer 感兴趣,也可以参考后面推荐的资源,扩展思路,挺有的。
建议你下载下来,先跑通,再慢慢改模型结构。哦对,记得留意一下输入文本的长度和词表维度,别一不小心 out of memory
了。
下载地址
用户评论