Modelpublic

kg-mc/bert-regresion-experiencia-laboral

sourceHugging Facemitupdated 10mo agoView on Hugging Face

1likes6downloads

Model Card

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

python

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))