Downloads · 30 days
0
t5-small-custom
sosolee/t5-small-custom
t5-small-custom is a machine learning model from sosolee. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Access
Public
Updated Sep 20, 2024
Repo size
—
Likes
0
Public
Repo makeup
Click a slice to open those files.
Other1.5 KB · 60%
At a glance
- Access
- Public
- Created
- Sep 20, 2024
- Updated
- Sep 20, 2024
- SHA
- cfae516d
Model card
From the Hugging Face model README
Model Card for t5_small Summarization Model
Model Details
- Model Name: t5-small
- Model Type: Text Summarization
- Architecture: Transformer-based model
- Number of Parameters: Approximately 60 million
Training Data
- Dataset: CNN/Daily Mail
- Data Source: Hugging Face Datasets
- Training Size: X samples (X는 실제 훈련 데이터 수로 교체)
- Data Characteristics: The dataset consists of news articles and their corresponding summaries.
Training Procedure
- Training Framework: Hugging Face Transformers
- Batch Size: 4
- Learning Rate: 2e-5
- Number of Epochs: 1
- Evaluation Metrics: ROUGE, BLEU
How to Use
- Dependencies:
- Install required libraries:
transformers,datasets
- Install required libraries:
- Example Usage:
from transformers import pipeline summarizer = pipeline("summarization", model="your_model_path") summary = summarizer("Your input text here.")
- Created
- Sep 20, 2024
- Updated
- Sep 20, 2024