심리 케어 챗봇(kogpt2, kobert) 구현해 배포해보자
[1] kogpt2 기반 심리 케어 챗봇 한줄 설명 KoGPT2 모델은 문장을 "생성"해내는 모델입니다. 심리 케어 목적의 챗봇을 구현하기 위해 입력 받은 내용에 대해 위로하거나 공감하거나 부드러운 표현으로 반응하고 문장을 생성해내도록 파인 튜닝을 진행했습니다. 학습 코드 링크입니다. 사용한 데이터셋 1. ai hub 제공, 웰니스 대화 스크립트 데이터셋 AI hub > 외부데이터 > KETI R&D데이터 >인식기술(언어지능) > 웰니스 대화 스크립트 데이터셋 2. @songys (송영숙님) 제공, 챗봇 데이터셋 두 데이터를 아래의 데이터 형식에 맞게 가공했습니다. 학습 환경 colab 기본 환경에서도 가능합니다만, 시간이 꽤 오래걸리는 편입니다! 1 epoch에 15~16분 정도 걸리는데 epoch 최..