Downloads · 30 days
16
5% of all-time downloads
textgain/News2Topic-T5-base
News2Topic-T5-base is a text generation model from textgain. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
- Model type: Text-to-Text Generation - Language(s) (NLP): English - License: MIT License - Finetuned from model: T5 Base Model (Google AI)
Downloads · 30 days
16
5% of all-time downloads
All-time downloads
311
Public
Repo size
1.8 GB
Likes
0
Public
Click a slice to open those files.
.bin892 MB · 100%
From the Hugging Face model README
The News2Topic T5-base model is designed for automatic generation of topic names from news articles or news-like text. It can be integrated into news aggregation platforms, content management systems, or used for enhancing news browsing and searching experiences by providing concise topics.
from transformers import pipeline
pipe = pipeline("text2text-generation", model="textgain/News2Topic-T5-base")
news_text = "Your news text here."
print(pipe(news_text))
The News2Topic T5-base model was trained on a 21K sample of the "Newsroom" dataset (https://lil.nlp.cornell.edu/newsroom/index.html) annotated with synthetic data generated by GPT-3.5-turbo
The model was trained for 3 epochs, with a learning rate of 0.00001, a maximum sequence length of 512, and a training batch size of 12.
BibTeX:
@article{Kosar_DePauw_Daelemans_2024,
title={Comparative Evaluation of Topic Detection: Humans vs. LLMs}, volume={13},
url={https://www.clinjournal.org/clinj/article/view/173}, journal={Computational Linguistics in the Netherlands Journal},
author={Kosar, Andriy and De Pauw, Guy and Daelemans, Walter},
year={2024},
month={Mar.},
pages={91–120} }