Lelapa AI Swahili Text Classification Task

Installation

Requirements

• Python 3.8.18
• CUDA 11.7 (optional)

Setup Environment

• Create a virtual environment with the specified Python version (recommended).
• Install the required packages from requirements.txt.

Setup Data

• Download these files and place them in the data/ folder:
  • Train.csv
  • Test.scv
  • SampleSubmission.csv

Note: All these steps must be carried out in the project's directory.

Train Model

python main.py

Notebooks

• Data Analysis and Exploration