This is a project to classify the Rails project GitHub issues into different categories using machine learning techniques. The project aims to analyze the issues reported on the Rails repository and classify them into different labels based on their descriptions. The classification model will help in organizing and prioritizing the issues reported by the community, making it easier for the maintainers to address them efficiently.
- Python
- Pytorch
- Transformers
- Hugging Face
- GitHub API
- ...
Make sure you have Python 3.12
installed on your system. You can install the required dependencies using the following command:
pip install -r requirements.txt
Generate the model using the github_issues_classification.ipynb
notebook.
Then run the project, using the following command:
python main.py
The detailed report of the project can be found in Report.md file.