kg-mc/bert-regresion-experiencia-laboral

BERT Regressor para Experiencia Laboral

Este modelo es un Fine-Tuning de dccuchile/bert-base-spanish-wwm-cased diseñado para extraer los años de experiencia requeridos de descripciones de puestos de trabajo en español.

Descripción

El modelo recibe un texto (descripción de empleo) y predice un número flotante que representa los años de experiencia solicitados.

• Entrada: Texto en español.
• Salida: Número flotante de años.

Rendimiento

• MAE (Error Medio Absoluto): 0.12 años
• RMSE: 0.14 años

Uso

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

model_id = "kg-mc/bert-regresion-experiencia-laboral"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)

texto = "Experiencia mínima de 5 años como desarrollador ABAP en entornos SAP S/4HANA."
inputs = tokenizer(texto, return_tensors="pt")

with torch.no_grad():
    outputs = model(**inputs)
    output = outputs.logits[0][0].item()
    print(round(output, 2))