SinclairSchneider's picture
Update README.md
0ad580b verified
|
raw
history blame
1.12 kB
metadata
license: mit
language:
  - de
pipeline_tag: text-classification
tags:
  - politics
datasets:
  - SinclairSchneider/trainset_political_text_yes_no_german

This classifier tells whether a German text is a political text or not. It is based on the model ikim-uk-essen/geberta-base and trained on the dataset SinclairSchneider/trainset_political_text_yes_no_german. The train script can be found under this link The model achieved an F1 score of 0.99 on the testset.

from transformers import pipeline

political_classifier = pipeline("text-classification", model="SinclairSchneider/german_politic_DeBERTa-v2-base")

political_classifier("Trump und Putin einigen sich auf begrenzte Waffenruhe")
[{'label': 'politic', 'score': 0.9999680519104004}]

political_classifier("Franck Ribéry und Arjen Robben feiern beim \"Beckenbauer Cup\" ihr Comeback beim FC Bayern. Sie präsentieren sich wie zu besten Zeiten und wecken eine große Sehnsucht beim Rekordmeister.")
[{'label': 'other', 'score': 0.9969837069511414}]