modelo-robertuito-nivelDeRiesgo5

modelo-robertuito-nivelDeRiesgo5 (epochs 5)

Descripción del modelo

Este es un modelo de clasificación de texto en español basado en RoBERTuito, fine-tuneado para la tarea de detección y clasificación de nivel de riesgo en textos en español para la evaluación docente de la Universidad Francisco de Paula Santander.

El modelo clasifica textos en 3 niveles de riesgo, permitiendo identificar de manera automatizada el grado de riesgo presente en un contenido textual.

• Idioma: Español (es)
• Licencia: Apache 2.0
• Modelo base: pysentimiento/robertuito-base-uncased
• Tarea: Clasificación de texto (multiclase)

Cómo usar el modelo

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

# Cargar modelo y tokenizer
model_name = "DevOB/modelo-robertuito-nivelDeRiesgo5"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)

# Texto de ejemplo
texto = "El profesor no tuvo métricas de evaluación durante todo el semestre."

# Tokenizar y predecir
inputs = tokenizer(texto, return_tensors="pt", truncation=True, padding=True, max_length=128)
with torch.no_grad():
    outputs = model(**inputs)
    logits = outputs.logits
    prediccion = torch.argmax(logits, dim=-1).item()

print(f"Nivel de riesgo predicho: {prediccion}")

Uso con pipeline

from transformers import pipeline

classifier = pipeline(
    "text-classification",
    model="DevOB/modelo-robertuito-nivelDeRiesgo5"
)

resultado = classifier("Texto a analizar aquí")
print(resultado)