Industry Classification

🚀 行業分類模型

本項目基於DistilBERT模型，可將商業描述分類到62個行業標籤中，為商業文本分類提供了有效的解決方案。

🚀 快速開始

本項目提供了基於DistilBERT的模型，可將商業描述分類為62個行業標籤之一。該模型在7000個印度公司的商業描述樣本及相關標籤上進行了訓練。

✨ 主要特性

• 基於DistilBERT模型，可將商業描述分類到62個行業標籤中。
• 使用7000個印度公司的商業描述樣本及相關標籤進行訓練。

📦 安裝指南

本項目未提供具體安裝命令，若要使用該模型，可參考如下依賴安裝：

pip install transformers

💻 使用示例

基礎用法

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 is an India-based non-banking financial company ... loan against property, management consultancy, personal loans and unsecured loans.")

'''Ouput'''
[{'label': 'Consumer Finance', 'score': 0.9841355681419373}]

🔧 技術細節

本模型基於DistilBERT，在7000個印度公司的商業描述樣本及相關標籤上進行訓練，可將商業描述分類到62個行業標籤中。

📄 許可證

本項目採用MIT許可證。

⚠️ 重要提示

訓練數據僅來自印度公司，可能存在一定的侷限性和偏差。