Karols's picture
Update README.md
51b3d03
metadata
language: de
tags:
  - text-classification
  - financial-sentiment-analysis
  - sentiment-analysis
datasets:
  - datasets/financial_phrasebank
metrics:
  - f1
  - accuracy
  - precision
  - recall
widget:
  - text: Der Nettoumsatz stieg um 30 % auf 36 Mio. EUR.
    example_title: Example 1
  - text: Der schwarze Freitag beginnt. Liste der Werbeaktionen in den Geschäften.
    example_title: Example 2
  - text: >-
      Die CDPROJEKT-Aktie verzeichnete den stärksten Rückgang unter den an der
      WSE notierten Unternehmen.
    example_title: Example 3

Finance Sentiment DE (base)

Finance Sentiment DE (base) is a model based on bert-base-german-cased for analyzing sentiment of German financial news. It was trained on the translated version of Financial PhraseBank by Malo et al. (20014) for 10 epochs on single RTX3090 gpu.

The model will give you a three labels: positive, negative and neutral.

How to use

You can use this model directly with a pipeline for sentiment-analysis:

from transformers import pipeline

nlp = pipeline("sentiment-analysis", model="bardsai/finance-sentiment-de-base")
nlp("Der Nettoumsatz stieg um 30 % auf 36 Mio. EUR.")
[{'label': 'positive', 'score': 0.9987998807375955}]

Performance

Metric Value
f1 macro 0.955
precision macro 0.960
recall macro 0.950
accuracy 0.966
samples per second 135.2

(The performance was evaluated on RTX 3090 gpu)

Changelog

  • 2023-09-18: Initial release

About bards.ai

At bards.ai, we focus on providing machine learning expertise and skills to our partners, particularly in the areas of nlp, machine vision and time series analysis. Our team is located in Wroclaw, Poland. Please visit our website for more information: bards.ai

Let us know if you use our model :). Also, if you need any help, feel free to contact us at [email protected]