ClueAI commited on
Commit
697eab6
1 Parent(s): 6fd73e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -4
README.md CHANGED
@@ -7,7 +7,7 @@ license: creativeml-openrail-m
7
 
8
  PromptCLUE:全中文任务零样本学习模型
9
 
10
- 这个模型是基于PromptCLUE-base进一步训练,以及更多任务上进行训练,是对PromptCLUE-base进一步升级
11
 
12
  在1000亿token中文语料上预训练,累计学习1.5万亿中文token,并且在数百种任务上进行Prompt任务式训练。针对理解类任务,如分类、情感分析、抽取等,可以自定义标签体系;针对多种生成任务,可以进行采样自由生成。
13
 
@@ -21,8 +21,8 @@ PromptCLUE:全中文任务零样本学习模型
21
  ```python
22
  # 加载模型
23
  from transformers import T5Tokenizer, T5ForConditionalGeneration
24
- tokenizer = T5Tokenizer.from_pretrained("ClueAI/PromptCLUE-base")
25
- model = T5ForConditionalGeneration.from_pretrained("ClueAI/PromptCLUE-base")
26
  ```
27
 
28
  使用模型进行预测推理方法:
@@ -37,7 +37,7 @@ def preprocess(text):
37
  def postprocess(text):
38
  return text.replace("_", "\n")
39
 
40
- def answer(text, sample=False, top_p=0.6):
41
  '''sample:是否抽样。生成任务,可以设置为True;
42
  top_p:0-1之间,生成的内容越多样'''
43
  text = preprocess(text)
@@ -51,6 +51,46 @@ def answer(text, sample=False, top_p=0.6):
51
  ```
52
 
53
  ### 示例输入
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
54
  #### 新闻分类(classify)
55
  ```bash
56
  Input:
 
7
 
8
  PromptCLUE:全中文任务零样本学习模型
9
 
10
+ 这个模型是基于PromptCLUE-base进一步训练,以及更多任务以及更多任务类型上进行训练,是对PromptCLUE-base进一步升级, 新增的任务类型有改写和问答等类型
11
 
12
  在1000亿token中文语料上预训练,累计学习1.5万亿中文token,并且在数百种任务上进行Prompt任务式训练。针对理解类任务,如分类、情感分析、抽取等,可以自定义标签体系;针对多种生成任务,可以进行采样自由生成。
13
 
 
21
  ```python
22
  # 加载模型
23
  from transformers import T5Tokenizer, T5ForConditionalGeneration
24
+ tokenizer = T5Tokenizer.from_pretrained("ClueAI/PromptCLUE-base-v1-5")
25
+ model = T5ForConditionalGeneration.from_pretrained("ClueAI/PromptCLUE-base-v1-5")
26
  ```
27
 
28
  使用模型进行预测推理方法:
 
37
  def postprocess(text):
38
  return text.replace("_", "\n")
39
 
40
+ def answer(text, sample=False, top_p=0.8):
41
  '''sample:是否抽样。生成任务,可以设置为True;
42
  top_p:0-1之间,生成的内容越多样'''
43
  text = preprocess(text)
 
51
  ```
52
 
53
  ### 示例输入
54
+ ### 改写(paraphrase)[smaple=True]
55
+ Input:
56
+ 生成与下列文字相同意思的句子:
57
+ 白云遍地无人扫
58
+ 答案:
59
+
60
+ Model output:
61
+ 白云散去无踪,没人扫。
62
+
63
+ Input:
64
+ 用另外的话复述下面的文字:
65
+ 怎么到至今还不回来,这满地的白云幸好没人打扫。
66
+ 答案:
67
+
68
+ Model output:
69
+ 满地白云无扫处
70
+
71
+ Input:
72
+ 改写下面的文字,确保意思相同:
73
+ 一个如此藐视本国人民民主权利的人,怎么可能捍卫外国人的民权?
74
+ 答案:
75
+
76
+ Model output:
77
+ 对一个如此藐视本国人民民主权利的人,怎么能捍卫外国人的民权?
78
+
79
+ ### 问答(qa)[smaple=False]
80
+ Input:
81
+ 根据问题给出答案:
82
+ 问题:手指发麻的主要可能病因是:
83
+ 答案:
84
+
85
+ Model output:
86
+ 神经损伤,颈椎病,贫血,高血压
87
+
88
+ 问答:
89
+ 问题:黄果悬钩子的目是:
90
+ 答案:
91
+
92
+ Model output:
93
+ 蔷薇目
94
  #### 新闻分类(classify)
95
  ```bash
96
  Input: