myml commited on
Commit
24f39fd
1 Parent(s): 0ccd07a

feat: 完善README

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md CHANGED
@@ -1,3 +1,47 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ widget:
4
+ - text: 只要关羽不捣乱,峡谷4V5也不怕
5
+ example_title: 只要关羽不捣乱,峡谷4V5也不怕
6
+ - text: 贵圈这场大戏真是活久见
7
+ example_title: 贵圈这场大戏真是活久见
8
+ - text: 职业操盘手不可外传的成交量买卖精髓,字字珠玑!
9
+ example_title: 职业操盘手不可外传的成交量买卖精髓,字字珠玑!
10
+ - text: 新疆呼图壁县举办第二届“渔之悦杯”垂钓比赛
11
+ example_title: 新疆呼图壁县举办第二届“渔之悦杯”垂钓比赛
12
  ---
13
+
14
+ # 中文新闻分类模型
15
+
16
+ ## 介绍
17
+
18
+ 基于 pytorch 使用[今日头条数据集](https://huggingface.co/datasets/fourteenBDr/toutiao)训练的中文新闻分类模型,用于学习中文文本分类使用,python 源码见 [main.py](./main.py)
19
+
20
+ ## 使用
21
+
22
+ ```python
23
+ from transformers import pipeline
24
+ classifier = pipeline("text-classification", model="myml/toutiao")
25
+ classifier("只要关羽不捣乱,峡谷4V5也不怕?")
26
+ # echo [{'label': 'LABEL_16', 'score': 0.9958072900772095}]
27
+ ```
28
+
29
+ ## 结果映射
30
+
31
+ ```txt
32
+ LABEL_0 民生 故事
33
+ LABEL_1 文化 文化
34
+ LABEL_2 娱乐 娱乐
35
+ LABEL_3 体育 体育
36
+ LABEL_4 财经 财经
37
+ LABEL_6 房产 房产
38
+ LABEL_7 汽车 汽车
39
+ LABEL_8 教育 教育
40
+ LABEL_9 科技 科技
41
+ LABEL_10 军事 军事
42
+ LABEL_12 旅游 旅游
43
+ LABEL_13 国际 国际
44
+ LABEL_14 证券 股票
45
+ LABEL_15 农业 三农
46
+ LABEL_16 电竞 游戏
47
+ ```