Clasificación de Industria
sampathkethineedi
Clasificación de texto
Modelo DistilBERT para clasificar una descripción de negocio en uno de los 62 tags de industria. Entrenado con 7000 muestras de Descripciones de Negocios y etiquetas asociadas de compañías en la India.
Como usar
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
tokenizer = AutoTokenizer.from_pretrained("sampathkethineedi/industry-classification")
model = AutoModelForSequenceClassification.from_pretrained("sampathkethineedi/industry-classification")
industry_tags = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
# Ejemplo de uso
industry_tags("Stellar Capital Services Limited es una empresa india de servicios financieros no bancarios ... préstamo contra propiedad, consultoría de gestión, préstamos personales y préstamos no asegurados.")
'''Salida'''
[{'label': 'Consumer Finance', 'score': 0.9841355681419373}]
Funcionalidades
- Clasificación de texto
- Transformadores
- PyTorch
- TensorFlow
- Clasificación multicategoría
Casos de uso
- Clasificación de descripciones de negocios
- Etiquetado de industrias en descripciones de negocios