This project aims to develop a chatbot using a Transformer model. It involves training a Transformer on conversational data and setting up an interactive chat interface.
transformer-chatbot/
│
├── data/
│ ├── raw_data/
│ ├── processed_data/
│ └── tokenizer/
│
├── models/
│ ├── transformer/
│ └── pretrained/
│
├── src/
│ ├── __init__.py
│ ├── train.py
│ ├── chat.py
│ ├── preprocess.py
│ └── utils.py
│
├── notebooks/
│
├── requirements.txt
│
└── README.md
- Clone the repository:
git clone <repository-url>
- Install dependencies:
pip install -r requirements.txt
- Run
python src/train.py
to train the model. - Run
python src/chat.py
to start the chatbot interface. - Use the Jupyter notebooks in the
notebooks/
directory for exploratory data analysis and testing.
Contributions to the project are welcome. Please follow the standard pull request process to make changes.
For any queries, please contact me.