NAICS GitHub 仓库分类器：基于 RoBERTa 的行业分类模型
原标题：aquiro1994/naics-github-classifier
AI 摘要
Hugging Face 上发布了 aquiro1994/naics-github-classifier 模型，这是一个基于 RoBERTa-large 微调的多类文本分类模型，用于将 GitHub 仓库分类到 19 个 NAICS 行业部门。该模型使用 6,588 个标注仓库训练，支持通过仓库名称、描述、主题和 README 进行预测。模型提供了详细的 Python 使用示例，并针对 Apple Silicon 设备优化了推理性能，在 MPS 上使用 fp16 精度可达到每秒 177 行的吞吐量。
正文节选
--- license: mit language: - en library_name: transformers tags: - text-classification - naics - industry-classification - github - roberta datasets: - custom metrics: - f1 - accuracy pipeline_tag: text-classification --- # NAICS GitHub Repository Classifier A fine-tuned RoBERTa-large model that classifies GitHub repositories into **19 NAICS (North American Industry Classification System)** industry sectors based on repository metadata. ## Model Description This model takes GitHub repository