cssupport
/

bert-news-class

Text Classification

Model card Files Files and versions

cssupport commited on Sep 27, 2023

Commit

7bea5bd

·

1 Parent(s): 0d706e8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -122,4 +122,4 @@ Pytorch and HuggingFace
 Misra, Rishabh. "News Category Dataset." arXiv preprint arXiv:2209.11429 (2022).
 Misra, Rishabh and Jigyasa Grover. "Sculpting Data for ML: The first act of Machine Learning." ISBN 9798585463570 (2021).
-Tandon, Karan. "This LLM is based on BERT (2018) a bidirectional Transformer. BERT was finetuned using AdamW with the help of NVIDIA AMP and trained in 45 minutes on one P6000 GPU. This model accepts news summary/news headlines/news article and classifies into one of 40 categories"

 Misra, Rishabh. "News Category Dataset." arXiv preprint arXiv:2209.11429 (2022).
 Misra, Rishabh and Jigyasa Grover. "Sculpting Data for ML: The first act of Machine Learning." ISBN 9798585463570 (2021).
+Tandon, Karan. "This LLM is based on BERT (2018) a bidirectional Transformer. **cssupport/bert-news-class** was finetuned using AdamW with the help of NVIDIA AMP and trained in 45 minutes on one P6000 GPU. This model accepts news summary/news headlines/news article and classifies into one of 40 categories"