A multilingual machine translation system built from scratch using custom datasets, tokenizers, and models. The goal is to build a flexible and modular translation pipeline that can be easily extended to new languages and models.
├── Datasets/ # Preprocessed or raw language datasets
├── Model/ # Model architecture and training scripts
├── Tokenizer/ # Tokenizer scripts or configs (e.g., SentencePiece, BPE)
├── Create_Dataset.ipynb # Jupyter notebook to create or preprocess datasets
└── README.md # Project documentation
- Multilingual support (e.g., English ↔ French, English ↔ Spanish)
- Custom tokenizer training (SentencePiece or other)
- Clean dataset preprocessing pipeline
- Modular, customizable model architecture
- Easy integration and extension
- Python 3.8+
- PyTorch
- Hugging Face Transformers
- SentencePiece
- Jupyter Notebook
- Create Dataset Open Create_Dataset.ipynb and generate your parallel corpora.
- Tokenizer Training Use scripts under Tokenizer/ to train or load your tokenizer.
- Model Training Modify and run training scripts inside the Model/ directory.
- Evaluation (To-do) Evaluate translation quality using BLEU, ROUGE, or other metrics.
Install dependencies:
pip install -r requirements.txt-
Create Dataset
OpenCreate_Dataset.ipynband generate your parallel corpora. -
Tokenizer Training
Use scripts underTokenizer/to train or load your tokenizer. -
Model Training
Modify and run training scripts inside theModel/directory. -
Evaluation
Evaluate translation quality using BLEU, ROUGE, or other metrics.
- Add CLI or Streamlit web demo
- Plug-and-play support for new language pairs
- Integration with external translation APIs for evaluation
Feel free to fork this project, raise issues, or submit PRs. Let’s build something awesome together.