Downloads · 30 days
35
0% of all-time downloads
owaiskha9654/Multi-Label-Classification-of-PubMed-Articles
Multi-Label-Classification-of-PubMed-Articles is a text classification model from owaiskha9654. Use it when you need a label for a piece of text. It is set up for transformers.
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…
Downloads · 30 days
35
0% of all-time downloads
All-time downloads
35.2K
Public
Parameters
108M
11.7 GB on disk
Likes
15
Public
Click a slice to open those files.
.bin433 MB · 50%
How the weights are stored.
F32108M · 100%
From the Hugging Face model README
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 BioBERT model to fine tune on PubMed MultiLabel classification Dataset.
Also tried RobertaForSequenceClassification and XLNetForSequenceClassification models for Fine-Tuning the Model on Pubmed MultiLabel Datset.
<img src="https://raw.githubusercontent.com/Owaiskhan9654/DigiGene/main/Paper%20Night%20Design.gif"> <img src="https://camo.githubusercontent.com/dd842f7b0be57140e68b2ab9cb007992acd131c48284eaf6b1aca758bfea358b/68747470733a2f2f692e696d6775722e636f6d2f52557469567a482e706e67">I have integrated Weight and Bias also for visualizations and logging artifacts and comparisons of Different models!
Multi Label Classification of PubMed Articles (Paper Night Event)
<img src="https://raw.githubusercontent.com/Owaiskhan9654/Gene-Sequence-Primer-/main/BioAsq.JPG">
- To get the API key, create an account in the website .
- Use secrets to use API Keys more securely inside Kaggle.
For more information on the attributes visit the Kaggle Dataset Description here.