gpt2_chatbot

☕ GPT2-Chatbot

GPT-2-Chatbot für tägliche Gespräche, trainiert auf den Datensätzen Daily Dialogue , Empathetic Dialogues , PERSONA-CHAT und Blended Skill Talk . Dieser Chatbot basiert auf dem GPT2-Modelltransformator mit einem Sprachmodellierungskopf an der Spitze.

⌛ Installation

Laden Sie das Modell aus dem AWS S3-Speicher herunter und führen Sie den folgenden Befehl aus:

git pull https://github.com/xcapt0/gpt2_chatbot.git
docker build -t gpt2_bot . 

? Verwendung

Führen Sie den Docker-Container aus:

docker run --rm -it gpt2_bot

Es gibt zwei verschiedene Möglichkeiten, den Chatbot zu verwenden: train und interact

Interaktionsmodus

Um den Chatbot zu starten, führen Sie den folgenden Befehl aus. Geben Sie --checkpoint Pfad zu Ihrem Modell an

python chatbot.py --mode interact --checkpoint path/to/model.h5

Zugmodus

Um das Modell zu trainieren, führen Sie den folgenden Befehl aus. Geben Sie bei Bedarf --checkpoint

python chatbot.py --mode train

Lizenz

Urheberrecht © 2022 xcapt0.
Dieses Projekt ist MIT-lizenziert.