Downloads · 30 days
4
13% of all-time downloads
pravin691983/fewshotlearning-bert-sm
fewshotlearning-bert-sm is a text classification model from pravin691983. Use it when you need a label for a piece of text. It is set up for transformers.
Downloads · 30 days
4
13% of all-time downloads
All-time downloads
30
Public
Parameters
109M
1.3 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt876 MB · 50%
From the Hugging Face model README
loss: 1.38671875
f1_macro: 0.1
f1_micro: 0.25
f1_weighted: 0.1
precision_macro: 0.0625
precision_micro: 0.25
precision_weighted: 0.0625
recall_macro: 0.25
recall_micro: 0.25
recall_weighted: 0.25
accuracy: 0.25
One of the potential business applications of few-shot text classification with the AG News dataset is in media and content companies. They could implement this technology to categorize news articles on world, sports, business, technology, and other topics with minimal labeled data. This few-shot model application would allow for more efficient management and retrieval of news content, improving user satisfaction with personalized news feed. Moreover, such a model will allow these companies to promptly adjust their classification to new categories or rapidly emerging topics in dynamic industries.
With a concern that the repetition of the source material may impair the perception of the results of my adaptation, I would prefer to avoid working with the same data I encountered during the course. Therefore, I would like to select a diverse text dataset where the number of the labelled examples available for each of the classes is limited. Additionally, in order to evaluate the effectiveness of the model, I would consider varying the domains and types of documents. The work will begin with the choice of the dataset, and the one I have selected is the AG’s News Corpus, which can be accessed on Hugging Face. In my study, I use this collection of news articles, divided into four primary classes: World, Sports, Business, and Sci/Tech. The sizes of the dataset are as follows: 30,000 training samples and 1,900 test samples for each of the classes.
Dataset size: 31.3 MB
Data Split: 127600 rows
Data Fields: