Wi/arxiv-distilbert-base-cased

Clasificación de Categorías de ArXiv con DistilBERT

Este modelo se ha afinado en un pequeño subconjunto del conjunto de datos de ArXiv para predecir la categoría de un documento dado.

Descargas el último mes: 214

Ejemplo de Uso

from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
import torch

# Cargar el modelo y el tokenizador
model_name = 'Wi/arxiv-distilbert-base-cased'
tokenizer = DistilBertTokenizer.from_pretrained(model_name)
model = DistilBertForSequenceClassification.from_pretrained(model_name)

# Preparar el texto
text = "Su texto aquí"
inputs = tokenizer(text, return_tensors="pt")

# Realizar la predicción
with torch.no_grad():
    logits = model(**inputs).logits

# Extraer la categoría
predicted_class_id = logits.argmax().item()

Este modelo aún no tiene suficiente actividad para ser desplegado en la API de Inferencia (sin servidor), pero puede ser desplegado en Puntos de Inferencia (dedicados).