Team Ai
Modelpublic

Ahmedjr/codebert-algorithm-tagger

sourceHugging Faceupdated 11mo agoView on Hugging Face
0likes21downloads
README.md58 linesDownload Raw Back to root
1---2language: en3tags:4- code5- algorithms6- competitive-programming7- multi-label-classification8- codebert9datasets:10- xCodeEval11metrics:12- f113- precision14- recall15library_name: transformers16pipeline_tag: text-classification17---18 19# CodeBERT Algorithm Tagger20 21A fine-tuned CodeBERT model for multi-label classification of algorithmic problems from competitive programming platforms like Codeforces.22 23## Model Description24 25This model predicts algorithmic tags/categories for competitive programming problems based on their problem descriptions and solution code.26 27**Supported Tags:**28- math29- graphs30- strings31- number theory32- trees33- geometry34- games35- probabilities36 37## Training Data38 39- **Dataset**: xCodeEval (Codeforces problems)40- **Training examples**: 2,147 problems (filtered for focus tags)41- **Test examples**: 531 problems42- **Source**: Problems and solutions from Codeforces platform43 44### Model Architecture45 46- **Input**: Concatenated problem description and solution code47- **Encoder**: CodeBERT (RoBERTa-based architecture)48- **Output**: 8-dimensional binary classification (one per tag)49 50## Usage51 52### Installation53 54```bash55pip install transformers torch56```57 58