MMLForge Search models, datasets, tasks… Start free
Home › Models › Summarization › nandakishormpai/t5-small-machine-articles-tag-generation
T

nandakishormpai/t5-small-machine-articles-tag-generation

Summarization·nandakishormpai· 454
transformers apache-2.0 Summarization

Machine Learning model to generate Tags for Machine Learning related articles. This model is a fine-tuned version of t5-small fine-tuned on a refined version of 190k Medium Articles dataset for generating Machine Learning article tags using the article textual content as input. While usually formulated as a multi-label classification problem, this model deals with tag generation as a text2text gen

Open in MLForge Sign up free Desktop app Source ↗
# pull & run locally
pip install mlforge-sdk && mlforge pull nandakishormpai/t5-small-machine-articles-tag-generation

Model details

Task
Summarization
Provider
nandakishormpai
Framework
transformers
Size
12 GB
License
apache-2.0
Downloads
454
Updated
2023-02-20

About nandakishormpai/t5-small-machine-articles-tag-generation

Machine Learning model to generate Tags for Machine Learning related articles. This model is a fine-tuned version of t5-small fine-tuned on a refined version of 190k Medium Articles dataset for generating Machine Learning article tags using the article textual content as input. While usually formulated as a multi-label classification problem, this model deals with tag generation as a text2text generation task (inspiration and reference: fabiochiu/t5-base-tag-generation).

Related Summarization

B facebook/bart-large-cnn Summarization ·406.3M params 1.6M 1.6K 🤗 HF D sshleifer/distilbart-cnn-12-6 Summarization 842.6K 322 🤗 HF P google/pegasus-xsum Summarization 241.9K 222 🤗 HF B philschmid/bart-large-cnn-samsum Summarization 144.8K 268 🤗 HF F human-centered-summarization/financial-summarization-pegasus Summarization ·568.8M params 137.4K 144 🤗 HF D sshleifer/distilbart-xsum-12-6 Summarization 99.5K 7 🤗 HF

Browse all Summarization models →