MPTI는 Meet Persona Through AI의 줄임말로 영화 속 캐릭터와의 대화형 챗봇 서비스입니다.
MPTI_Sherlock은 DialoGPT를 사용하여, MPTI_Ironman은 GPT2를 사용하여 만들었습니다.
서비스에 대한 자세한 내용은 그림 위의 링크를 참고해주세요.
학습에 사용된 데이터는 대본을 바탕으로 구성되었으며, 상업적으로 시용할 의도가 전혀 없음을 밝힙니다:)
각각의 챗봇에 사용한 데이터의 형식이 서로 다릅니다. 데이터의 형식은 개별 README를 통해 확인할 수 있습니다.
모델의 용량 문제로 인해, 16GB 이상의 GPU 환경이나 Colab에서 GPU를 사용하기를 적극 권장합니다. Colab 환경의 경우, [런타임] - [런타임 유형 변경] - [GPU]로 설정을 바꿔주세요.
git clone https://github.com/TobigsConf-1415/MPTI.git
cd MPTI
pip install -r requirements.txt
각 모델을 아래의 순서대로 학습시켜주세요
cd DialoGPT
python main.py
cd TransferTransfo
python train.py
python demo.py
Sherlock | Iron Man |
---|---|
![]() |
![]() |
- Golsun, DialogRPT, 2020, https://github.com/golsun/DialogRPT
- HuggingFace, transfer-learning-conv-ai, 2020, https://github.com/huggingface/transfer-learning-conv-ai
- Wolf, T., Sanh, V., Chaumond, J., & Delangue, C. (2019). Transfertransfo: A transfer learning approach for neural network based conversational agents.
- Zhang, Y., Sun, S., Galley, M., Chen, Y. C., Brockett, C., Gao, X., ... & Dolan, B. (2019). Dialogpt: Large-scale generative pre-training for conversational response generation.
- Thomas Wolf, How to build a State-of-the-Art Conversational AI with Transfer Learning, 2019, https://medium.com/huggingface/how-to-build-a-state-of-the-art-conversational-ai-with-transfer-learning-2d818ac26313
- https://towardsdatascience.com/make-your-own-rick-sanchez-bot-with-transformers-and-dialogpt-fine-tuning-f85e6d1f4e30
![]() Seyoung Jung |
![]() Jaeyoon Jeong |
![]() Taehee Kim |
![]() Sumin Lee |
![]() Junhyeok Cho |
![]() JinKyoung Hwangbo |