industry-classification-api
sampathkethineedi
Clasificación de texto
Modelo BERT para clasificar una descripción comercial en una de 62 etiquetas de industria. Entrenado con 7000 muestras de Descripciones Comerciales y etiquetas asociadas de empresas en India.
Como usar
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
tokenizer = AutoTokenizer.from_pretrained("sampathkethineedi/industry-classification")
model = AutoModelForSequenceClassification.from_pretrained("industry-classification")
industry_tags = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
industry_tags("Stellar Capital Services Limited es una empresa financiera no bancaria con sede en India... préstamo contra propiedad, consultoría de gestión, préstamos personales y préstamos no garantizados.")
'''Ouput'''
[{'label': 'Finanzas al Consumidor', 'score': 0.9841355681419373}]
Funcionalidades
- Clasificación de texto
- Transformers
- PyTorch
- JAX
- Clasificación multi-etiqueta
- Inference Endpoints
Casos de uso
- Clasificación de descripciones comerciales en etiquetas de industria
- Identificación de etiquetas de negocios en empresas indias