metadata

license: apache-2.0
language:
  - en
metrics:
  - accuracy
pipeline_tag: text-classification
tags:
  - news
widget:
  - text: >-
      Researchers have made significant progress in the development of a new
      treatment for a rare genetic disorder. Early trials of the treatment have
      shown promising results, with patients experiencing improvements in their
      symptoms and quality of life. This breakthrough offers hope to individuals
      and families affected by the condition, bringing them closer to a
      potential cure

Fine-Tuned BERT News Classifier

Overview

The Fine-Tuned BERT News Classifier is a natural language processing (NLP) model built upon the BERT architecture. It is specifically designed for news classification, providing a softmax output where a value of 1 indicates positive news and 0 indicates negative news sentiment. This model is trained to understand and categorize news articles, assisting in tasks such as sentiment analysis and news aggregation.

Usage Instructions

Import Necessary Libraries

import tensorflow_text as text
import tensorflow as tf

Load The Model

from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("weightedhuman/fine-tuned-bert-news-classifier")

Make Predictions

examples = "Community Gardens Flourish, Bringing Fresh Produce and Unity to Neighborhoods"


serving_results = model \
                .signatures['serving_default'](tf.constant(examples))


serving_results = tf.sigmoid(serving_results['classifier'])
    
serving_results_np = serving_results.numpy()

for i in range(len(serving_results_np)):

    output_value = serving_results_np[i][0]

print(output_value)