语言: "英文" 缩略图: "https://huggingface.co/sampathkethineedi" 标签:

• distilbert
• pytorch
• tensorflow
• 文本分类
• 行业
• 商业
• 描述
• 多类别
• 分类 许可证: "mit" 推理: 不支持

行业分类模型

模型描述

基于DistilBERT的模型，用于将商业描述分类为62个行业标签之一。
训练数据包含7000个印度公司的商业描述及对应标签样本。

使用方法

提供PyTorch和TensorFlow版本

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)
industry_tags("Stellar Capital Services Limited是一家总部位于印度的非银行金融机构...提供房产抵押贷款、管理咨询、个人贷款和无担保贷款等服务。")

'''输出'''
[{'label': '消费金融', 'score': 0.9841355681419373}]

局限性及偏差

训练数据仅包含印度企业样本