A simple chatbot project implemented using PyTorch, designed to provide a basic understanding of building chatbots with neural networks.
- Training Script: Train the chatbot using PyTorch with a feedforward neural network.
- Inference Script: Interact with the trained chatbot in a simple command-line interface.
- Modular Structure: Organized project structure with separate modules for dataset, model, and utilities.
- Requirements File:
requirements.txt
file included for easy installation of dependencies.
- Python 3.x
- PyTorch
- torchvision
- Other dependencies specified in
requirements.txt
-
Clone the repository:
git clone https://github.com/https://github.com/DaveSimoes/Web-Scraping.git/chatbot-pytorch.git
-
Install the required dependencies:
pip install -r requirements.txt
-
Train the chatbot:
python train_chatbot.py
-
Interact with the chatbot:
python chatbot_app.py
-
Follow the on-screen instructions to chat with the trained chatbot.
- Adjust hyperparameters (e.g.,
input_size
,hidden_size1
,hidden_size2
) in the training script according to your needs. - Customize the
intents.json
file in thedata
directory to define different intents and responses for your chatbot.
This project is licensed under the MIT License - see the LICENSE file for details.