Command Palette
Search for a command to run...
Tag Recommendation for Online Q&A Communities based on BERT Pre-Training Technique
Navid Khezrian Jafar Habibi Issa Annamoradnejad

Abstract
Online Q&A and open source communities use tags and keywords to index, categorize, and search for specific content. The most obvious advantage of tag recommendation is the correct classification of information. In this study, we used the BERT pre-training technique in tag recommendation task for online Q&A and open-source communities for the first time. Our evaluation on freecode datasets show that the proposed method, called TagBERT, is more accurate compared to deep learning and other baseline methods. Moreover, our model achieved a high stability by solving the problem of previous researches, where increasing the number of tag recommendations significantly reduced model performance.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| multi-label-text-classification-on-freecode | TagCNN | F1-score: 45.3 |
| multi-label-text-classification-on-freecode | TagBERT | F1-score: 46 |
| multi-label-text-classification-on-freecode | TagMulRec | F1-score: 36.4 |
| multi-label-text-classification-on-freecode | FastTagRec | F1-score: 33.2 |
| multi-label-text-classification-on-freecode | EnTagRec | F1-score: 36 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.