Downloads · 30 days
7
10% of all-time downloads
Stickmu/ThunderstormBERT-de-v1
ThunderstormBERT-de-v1 is a text classification model from Stickmu. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
ThunderBERT-de-v1 is a BERT-based language model for multi-class classification of German-language texts that describe thunderstorm events. Its purpose is to automatically detect and differentiate between varying inte…
Downloads · 30 days
7
10% of all-time downloads
All-time downloads
70
Public
Parameters
279M
1.1 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors1.1 GB · 98%
From the Hugging Face model README
ThunderBERT-de-v1 is a BERT-based language model for multi-class classification of German-language texts that describe thunderstorm events.
Its purpose is to automatically detect and differentiate between varying intensities of thunderstorm-related content in historical and contemporary German sources.
The model was developed and trained during Phase B of the Thunderstorm Project at the University of Freiburg.
de)bert-base-german-casedGW0: no thunderstormGW1: light thunderstormGW2: strong thunderstormGW3: severe thunderstormThe model uses a SentencePiece-based tokenizer configuration (spm.model, tokenizer_config.json, etc.).
For compatibility with the training setup, it may be necessary to load the tokenizer with use_fast=False.
https://huggingface.co/<namespace>/ThunderBERT-de-v1<namespace> with your actual HF username or organization)This model is designed for automatic classification of thunderstorm intensities in German-language texts.
It assigns each input text (sentence, passage, or short paragraph) to one of four categories:
| Label | Description |
|---|---|
| GW0 | no thunderstorm |
| GW1 | light thunderstorm |
| GW2 | strong thunderstorm |
| GW3 | severe thunderstorm |
Typical applications include:
Can be integrated into:
Not intended for:
from transformers import AutoTokenizer, AutoModelForSequenceClassification
model_id = "<namespace>/ThunderBERT-de-v1" # e.g. "Stickmu/ThunderBERT-de-v1"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)
text = "Am Abend entluden sich schwere Gewitter über dem Schwarzwald."
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
predicted_class_id = int(outputs.logits.argmax(dim=-1))
print(model.config.id2label[predicted_class_id])