Skip to content

Using BERT and TextCNN to solve the preblom of Multi-Label Text Classfication

Notifications You must be signed in to change notification settings

FadedCosine/AIFuture-Multi-Label-Text-Classfication

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 

Repository files navigation

未来杯“战疫”特别赛事 作品

概述

赛事方提供了基于十万余条已经标注负面情绪标签的训练集,希望通过建立并训练模型,使用该模型为测试集数据标注情绪标签的方式,发现潜在的心理疏导需求,从而辅助人们解决抗疫过程中出现的心理健康问题。

数据集中的标签包括六个,分别是:'恐慌', '悲伤', '惧怕', '愤怒', '无助', '焦虑';每一条文本可能带有三个及一下的标签,故实际上是一个中文文本情感多标签分类的任务。

我分别搭建了BERT和TextCNN两个模型,其中BERT得分0.481875,TextCNN得分0.432720,最终排名第四。

About

Using BERT and TextCNN to solve the preblom of Multi-Label Text Classfication

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages