Model Card for Model ID

Model Details

Model Description

news_classifier_model is a text classification model built using the pre-trained distilbert-base-uncased architecture. It is designed to classify news excerpts into five categories: Business, Opinion, Political Gossip, Sports, and World News.

The model is based on DistilBERT, a lightweight and faster version of BERT that retains strong language understanding capabilities while using fewer computational resources. It was fine-tuned on a dataset of Sri Lankan news articles, allowing it to learn patterns and context specific to news content.

By leveraging contextual understanding of text, the model can accurately analyze a given news excerpt and assign it to the most relevant category.

Intended Uses & Limitations

Intended Uses

News article classification Content organization and tagging Integration into web applications for automated categorization

Limitations

May not generalize well to news outside the training domain Performance can drop on very short or unclear text Not suitable for text generation tasks

Model Sources [optional]

  • Repository: [More Information Needed]
  • Paper [optional]: [More Information Needed]
  • Demo [optional]: [More Information Needed]

Uses

Direct Use

[More Information Needed]

Downstream Use [optional]

[More Information Needed]

Out-of-Scope Use

[More Information Needed]

Bias, Risks, and Limitations

[More Information Needed]

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Training Details

Training Data

[More Information Needed]

Training Procedure

Preprocessing [optional]

[More Information Needed]

Training Hyperparameters

  • Training regime: [More Information Needed]

Speeds, Sizes, Times [optional]

[More Information Needed]

Evaluation

Testing Data, Factors & Metrics

Testing Data

[More Information Needed]

Factors

[More Information Needed]

Metrics

[More Information Needed]

Results

[More Information Needed]

Summary

Model Examination [optional]

[More Information Needed]

Environmental Impact

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).

  • Hardware Type: [More Information Needed]
  • Hours used: [More Information Needed]
  • Cloud Provider: [More Information Needed]
  • Compute Region: [More Information Needed]
  • Carbon Emitted: [More Information Needed]

Technical Specifications [optional]

Model Architecture and Objective

[More Information Needed]

Compute Infrastructure

[More Information Needed]

Hardware

[More Information Needed]

Software

[More Information Needed]

Citation [optional]

BibTeX:

[More Information Needed]

APA:

[More Information Needed]

Glossary [optional]

[More Information Needed]

More Information [optional]

[More Information Needed]

Model Card Authors [optional]

[More Information Needed]

Model Card Contact

[More Information Needed]

Downloads last month
55
Safetensors
Model size
67M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Spaces using Akilashamnaka12/news_classifier_model 3

Paper for Akilashamnaka12/news_classifier_model