감정분류 모델을 만들어봤습니다.

2진 분류가 아닌 특정감정으로 분류하는 모델을 만들었는데

많은 곳에서 인용한 거라

데이터를 모으는 과정 외에는 사실상 짜집기 입니다.

하지만 이런 자료도 필요하신 분이 계시겠죠!

https://github.com/emotion-classifier/emotion_classifier

 

emotion-classifier/emotion_classifier

Use bert model for claffify the emotion of sentence - emotion-classifier/emotion_classifier

github.com

github주소 입니다.

Naive bayes

SVM

biLSTM

bert

네가지 모델을 돌려보았고

SVM과 bert가 80%대의 정확도를 보였습니다.

자세한 내용은 accuracy report폴더의 엑셀파일을 참고해주세요

데이터가 필요하신 분도 계실 거 같아 데이터도 github에 올려두었습니다!

+ Recent posts