| | --- |
| | license: mit |
| | widget: |
| | |
| | - text: 从北京到莫斯科 |
| | example_title: 从北京到莫斯科 |
| | - text: 色卡里的春天 |
| | example_title: 色卡里的春天 |
| | - text: 中俄联合声明 释放了几个信号 |
| | example_title: 中俄联合声明 释放了几个信号 |
| | - text: 鸡鸣寺游客爆满设反悔门引导离寺 |
| | example_title: 鸡鸣寺游客爆满设反悔门引导离寺 |
| | datasets: |
| | - fourteenBDr/toutiao |
| | language: |
| | - zh |
| | pipeline_tag: text-classification |
| | --- |
| | |
| | # 中文新闻分类模型 |
| |
|
| | ## 介绍 |
| |
|
| | 基于 pytorch 使用[今日头条数据集](https://huggingface.co/datasets/fourteenBDr/toutiao)训练的中文新闻分类模型，用于学习中文文本分类使用，python 源码见 [main.py](./main.py) |
| |
|
| | ## 使用 |
| |
|
| | ```python |
| | from transformers import pipeline |
| | classifier = pipeline("text-classification", model="myml/toutiao") |
| | classifier("只要关羽不捣乱，峡谷4V5也不怕？") |
| | # echo [{'label': 'LABEL_16', 'score': 0.9958072900772095}] |
| | ``` |
| |
|
| | ## 结果映射 |
| |
|
| | ```txt |
| | LABEL_0 民生 故事 |
| | LABEL_1 文化 文化 |
| | LABEL_2 娱乐 娱乐 |
| | LABEL_3 体育 体育 |
| | LABEL_4 财经 财经 |
| | LABEL_6 房产 房产 |
| | LABEL_7 汽车 汽车 |
| | LABEL_8 教育 教育 |
| | LABEL_9 科技 科技 |
| | LABEL_10 军事 军事 |
| | LABEL_12 旅游 旅游 |
| | LABEL_13 国际 国际 |
| | LABEL_14 证券 股票 |
| | LABEL_15 农业 三农 |
| | LABEL_16 电竞 游戏 |
| | ``` |
