[NLP]Chatbot

GomHyeok·2022년 5월 30일

NLP

목록 보기
2/2

Chatbot

NLP에서 학습한 내용을 Project에서 실제로 구현하기 위해 실시하였다.
해당 관련 코드는 github에서 확인 가능하다.


Chatbot_v1.0

Requirements

  • transformer install
pip install transformers

BERT모델을 기반으로 하기 때문에 transformers의 설치가 필요하다.

  • 욕설 Data
git clone https://github.com/LDNOOBW/List-of-Dirty-Naughty-Obscene-and-Otherwise-Bad-Words
git clone https://github.com/doublems/korean-bad-words	
git clone https://github.com/organization/Gentleman/

욕설이 포함된 문장을 처리하기 위한 욕설 Data


v1.0의 주요 기능으로
1. CLS Tokenizing과 cosine_similarity를 활용한 2개 문장 사이의 유사도 추출이 있다.
2. 욕설이 포함되어있는지 검사 및 욕설 포함시 문장 변경이 있다.


Chatbot_v2.0

Requirements

  • v1.0과 요구사항은 동일하다.
  • train.py : 모델을 학습시키는 파일
  • maketoken.py : Data를 tag별로 CLS Tokenizing 하여 pickle화 하여 저장
  • utils.py : 기타 함수
  • chatbot.py : pth파일(모델의 학습 결과 저장 파일)을 통하여 실제 기능을 구현하는 file
  • ChatbotData.csv : 모델의 학습에 사용되는 Data

v2.0의 주요 기능으로
1. tag를 기준으로 Dataset을 다시 만든다
2. 사용자의 입력이 어떤 tag를 가지고 있는지 추측
3. 추측된 tag와 동일한 Dataset의 CLS만 확인하여 답변의 정확도를 높인다.

profile
github : https://github.com/GomHyeok/

0개의 댓글