Skip to content
Simple rrn chatbot
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data
.gitignore
LICENSE
README.md
requierements.txt
train_rnn.py
train_seq2seq.py

README.md

rnn_chatbot

Simple rrn chatbot

Instalación

Para obtener el código puede clonarlo o bajarlo

Clonar

git clone git@github.com:ivanvladimir/rnn_chatbot.git

Bajarlo

Bajarlo de a aquí

Descomprimir

Habilitar virtualenv

Instalar virtualenv

sudo apt install virtualenv

Activar virtualenv en el directorio con el código

source rnn_chatbot/bin/activate

Instalar los requierements

En el directorio del código hacer

pip install -r requierements.txt

Ejecución

Existen tres modos:

  • Entrenamiento (train)
  • Prueba iterativa (test)
  • Limpieza (clean)

Entrenamiento

El sistema aprende a construir respuestas

python3 train_rnn.py data/conversations.fruta.json

El sistema comienza a iterar sobre los datos para determinar los parámetros (pesos) y genera los siguientes archivos:

  • Datos: matrices de entrenamiento y prueba (prueba ignorada por ahora)
  • Vocabulario: lista de indices a palabras, ya que el modelo no entiende palabras todo se pasa a indices y nos sirve para recuperar a que se refiere cada indice
  • Modelo: pesos del modelo

Una vez que se generan estos archivos el sistema los lee en lugar de generarlos y se pude iterar multiples veces

Prueba iterativa

En este modo se lanza un pequeño chatbot que responde de forma automática

python3 train_rnn.py data/conversations.fruta.json --mode test

para salir teclear exit o ctrl+d.

Limpieza

Usar con mucha precaución, borra los modelos y archivos auxiliares y se pierde el modelo. Excelente para cuando se quiera probar otros argumentos de la línea de comandos.

python3 train_rnn.py data/conversations.fruta.json --mode clean
You can’t perform that action at this time.