📽MPTI💬

MPTI는 Meet Persona Through AI의 줄임말로 영화 속 캐릭터와의 대화형 챗봇 서비스입니다.

MPTI_Sherlock은 DialoGPT를 사용하여, MPTI_Ironman은 GPT2를 사용하여 만들었습니다.

서비스에 대한 자세한 내용은 그림 위의 링크를 참고해주세요.

Data

학습에 사용된 데이터는 대본을 바탕으로 구성되었으며, 상업적으로 시용할 의도가 전혀 없음을 밝힙니다:)

각각의 챗봇에 사용한 데이터의 형식이 서로 다릅니다. 데이터의 형식은 개별 README를 통해 확인할 수 있습니다.

• Sherlock
• Iron Man

Usage

모델의 용량 문제로 인해, 16GB 이상의 GPU 환경이나 Colab에서 GPU를 사용하기를 적극 권장합니다. Colab 환경의 경우, [런타임] - [런타임 유형 변경] - [GPU]로 설정을 바꿔주세요.

1. Installation

git clone https://github.com/TobigsConf-1415/MPTI.git
cd MPTI
pip install -r requirements.txt

2. Train

각 모델을 아래의 순서대로 학습시켜주세요

cd DialoGPT
python main.py

cd TransferTransfo
python -m spacy download en
python train.py

3. Inference

python demo.py

Result

Demo

Sherlock	Iron Man

Reference

• 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

Contributor 🕵️‍♂️

Seyoung Jung

Jaeyoon Jeong

Taehee Kim

Sumin Lee

Junhyeok Cho

JinKyoung Hwangbo