Importante : Este repositório foi desenvolvido como uma forma de treinar um rede neural LSTM, mas precisamente uma rede sequence to sequence afim de treinar um bot de proposito geral, e adiquirir experiencia com Deep Learning.
Setup
git clone https://github.com/erickrribeiro/LSTM-Chatterbot.git
cd LSTM-Chatterbot
Durante o decorrer das épocas no treinamento da rede neural este projeto fará uma serie de backups, para garantir que a qualquer momento será possível parar o treinamente, avaliar o desempenho da rede, e continuar e onde parou caso seja necessário.
Para isso é necessário executar o seguinte script:
./setup
Como resultado será criado o seguinte esquema de pastas, onde os metadados da rede serão armazenados.
└── experiment --
├── data --
├── nn_models --
└── results --
Resultado:
hello baby - hello
how old are you ? - twenty .
i am lonely - i am not
nice - you ' re not going to be okay .
so rude - i ' m sorry .
Papers
Arquitetura
Run
python train.py
Testar o chatbot pra um conjunto de frases pré-definidas:
python test.py
Playground:
python chat.py
Todos os parametros de configuração esão em app/configs/config.py
Requirements