Simple Chatbot using PyTorch - 3 | Creating and training the Neural Network model.

In this video, we are going to define the neural network model and also train it.

The training data was created in the last video and in this video we create the model for it. We define a two layer simple neural network using relu as the activation function. The input, output and the hidden dimensions are kept as 48 , 6 , 16.

The video also ends with showing the decreasing loss of the model.

Code:

Instagram: