Create README.md

# Migration Stance Classification Model

This model categorizes the stance expressed in text regarding migration into three categories: **Positive**, **Negative**, or **Neutral**.

## Model Description

This model is fine-tuned to detect stance in migration-related content. It can identify whether a text expresses support for migration (positive), opposition to migration (negative), or presents factual/neutral information.

**Key Features:**
- **3-class classification**: POSITIVE, NEGATIVE, NEUTRAL
- **Domain-specific**: Optimized for migration and immigration discourse in Slovakia

## Labels

- **0 - NEGATIVE**: Text expressing opposition, concerns, or negative views about migration
- **1 - NEUTRAL**: Factual statements, balanced reporting, or no clear stance
- **2 - POSITIVE**: Text expressing support, benefits, or positive views about migration

## Usage

### Basic Usage
```python
from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch

# Load model and tokenizer
model_name = "MIMEDIS/stance-model"
model = AutoModelForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

# Prepare input
text = "Migrácia obohacuje našu spoločnosť o nové perspektívy a kultúry."
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)

# Get predictions
with torch.no_grad():
outputs = model(**inputs)
predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
predicted_class = torch.argmax(predictions, dim=-1).item()

# Map class to label
labels = {0: "NEGATIVE", 1: "NEUTRAL", 2: "POSITIVE"}

print(f"Text: {text}")
print(f"Predicted stance: {labels[predicted_class]}")
print(f"Confidence: {predictions[0][predicted_class]:.4f}")
print(f"All probabilities: NEGATIVE={predictions[0][0]:.4f}, NEUTRAL={predictions[0][1]:.4f}, POSITIVE={predictions[0][2]:.4f}")

README.md

@@ -0,0 +1,15 @@
+---
+license: apache-2.0
+datasets:
+- MIMEDIS/migration-stance-content
+language:
+- sk
+base_model:
+- gerulata/slovakbert
+pipeline_tag: text-classification
+tags:
+- stance-classification
+- text-classification
+- slovakbert
+- slovak
+---