본문으로 건너뛰기

3️⃣ 삼행시의 달인

프로젝트 진행 과정​

전체 파이프라인(streamlit을 통해 시현)​

Untitled

1. 라벨링​

  • KLUE RoBERTa base 이용해 감정 분류
  • KoBERT로 생성한 에타 데이터에 어색한 문장이 존재하여 RoBERTa-base 이용해 어색한 문장, 정상적인 문장 학습하여 일부 데이터 삭제

그림1.png

2. 모델링​

  • KoGPT-2 Trinity를 기본으로 Conditional GPT를 이용해 감정분류 logit 값을 main/sub condition으로 줌

Untitled

  • 삼행시의 각 문장이 더 자연스럽게 이어질 수 있도록 어미 변경 및 접속사 추가

그림3.png

3. 웹​

  • 웹 어플리케이션인 streamlit을 통해 구현
  • Huggingface에 모델을 올려 사용자들이 사용할 수 있도록 함

KoGPT-Conditional-Generation - a Hugging Face Space by snoop2head

프로젝트 결과​

Streamlit을 통한 시현 결과​

  1. 주요 감정 두가지를 선택

  2. 삼행시를 만들 단어 입력

    —> 감정을 반영한 삼행시 출력

Untitled