/koGPT2_Chatbot_ToyExample

koGPT2를 활용해 챗봇 만들기(가장 쉬운 예제)

Primary LanguageJupyter Notebook

koGPT2 + pytorch 챗봇 만들기 실습

  • 주제: koGPT2와 pytorch를 이용해 한국어 챗봇을 간단하게 만들어보기
  • 출처: 위키독스
  • 내용:
    • 사전학습 모델인 koGPT2를 베이스로 별도의 데이터(ChatBotData.csv)를 추가 학습해 챗봇을 만든 후 테스트
    • 코드: 위 출처의 코드를 이용해 학습 실습하되 전처리 및 학습 관련 함수는 모두 모듈화
  • 주의
    • CPU로 실행할 경우 매우 느림
    • 챗봇 학습 모델은 파일 용량이 500mb로 깃헙 업로드 불가
    • 전처리 & 학습 관련 함수 구조 + 이론 설명은 추후 업로드