Chatbot with PyTorch 🤖

📖 Preproccesing

Much of this work was inspired by this tutorials. For the preprocessing of the training text, the techniques of tokenization, bag of words and steeming have been used. The input to the network will be a vector with 0's and 1's (product of preprocessing).

Pipeline of the preproccesing

🧠 Model

The architecture used was a feedforward network (very simple). The network receives a input vector (length equal to size of bag of words). Subsequently, it shows the user a random response (from the json file) based on that category.

Part of the network code is shown below.

class Net(nn.Module):
    def __init__(self, input_size, hidden_size, output_size):
        super(Net, self).__init__()
        self.linear1 = nn.Linear(input_size, hidden_size)
        self.linear2 = nn.Linear(hidden_size, hidden_size)
        self.linear3 = nn.Linear(hidden_size, output_size)
        self.relu = nn.ReLU()

    def forward(self, x):
        x = self.relu(self.linear1(x))
        x = self.relu(self.linear2(x))
        output = self.linear3(x)
        return output

Architecture of the neural network

For train the network, run:

python train.py

Customize ⚙️

The training dataset used is in the intents.json file, which contains texts in Spanish that will serve as quick responses from our chatbot (A tokenizer and steemer focused on the Spanish language must be used). If you want the chatbot to have more elaborate responses, you can expand the categories and quick responses. Just define a new tag, possible patterns, and possible responses for the chat bot. You have to re-run the training whenever this file is modified.

{
    "intents": [
      {
        "tag": "saludo",
        "patterns": [
          "Hola",
          "Qué tal?",
          "Hay alguien allí?",
          "Buenos dias",
          "Buen dia"
        ],
        "responses": [
          "Hola :^)",
          "Hola, gracias por la visita",
          "Hola, qué puedo hacer por ti?",
          "Hola, cómo puedo ayudarte?"
        ]
      },
      {
        "tag": "despedida",
        "patterns": ["Adiós", "Nos vemos luego", "Hasta la próxima"],
        "responses": [
          "Hasta luego, gracias por tu visita",
          "Ten un buen dia",
          "Chau, espero verte de nuevo."
        ]
      },
    ...
  ]
}

▶ Demo

If you want to run application, simply install all the dependencies of the requirements.txt and execute the following line.

python app.py

One windows will open immediately, as shown below.

👨‍💻 Maintainers

• Cristhian Wiki, Github: HiroForYou Email: csanchezs@uni.pe