Owaiskhan9654/Multi-Label-Classification-of-Pubmed-Articles
The traditional machine learning models give a lot of pain when we do not have sufficient labeled data for the specific task or domain we care about to train a reliable model. Transfer learning allows us to deal with these scenarios by leveraging the already existing labeled data of some related task or domain. We try to store this knowledge gained in solving the source task in the source domain and apply it to our problem of interest. In this work, I have utilized Transfer Learning utilizing BertForSequenceClassification model. Also tried RobertaForSequenceClassification and XLNetForSequenceClassification models for Fine-Tuning the Model.
This project helps medical researchers, librarians, or data scientists automatically categorize PubMed articles. It takes the text of a research article and assigns multiple relevant tags or labels, even when you have limited pre-labeled data for your specific area. This allows for more efficient organization and retrieval of medical literature.
No commits in the last 6 months.
Use this if you need to classify biomedical articles into multiple categories but lack a large, custom-labeled dataset to train a model from scratch.
Not ideal if your classification task is outside the biomedical domain or if you already have extensive, high-quality labeled data for your specific needs.
Stars
21
Forks
6
Language
Jupyter Notebook
License
Apache-2.0
Category
Last pushed
Aug 27, 2025
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/Owaiskhan9654/Multi-Label-Classification-of-Pubmed-Articles"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
castorini/hedwig
PyTorch deep learning models for document classification
AnubhavGupta3377/Text-Classification-Models-Pytorch
Implementation of State-of-the-art Text Classification Models in Pytorch
inspirehep/magpie
Deep neural network framework for multi-label text classification
kk7nc/Text_Classification
Text Classification Algorithms: A Survey
InseeFrLab/torchTextClassifiers
A unified framework for text classification in PyTorch.