YanshekWoo commited on
Commit
ab93ca6
1 Parent(s): 824e6bf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -1
README.md CHANGED
@@ -12,4 +12,13 @@ We utilize 4 Chinese dialogue datasets from [LUGE](https://www.luge.ai/#/)
12
  | Chinese Persona Chat (CPC) | 23,000 | Open |
13
  | LCCC | 11,987,759 | Open |
14
  | Emotional STC (ESTC) | 899,207 | Open |
15
- | KdConv | 3,000 | Movie, Music, Travel |
 
 
 
 
 
 
 
 
 
 
12
  | Chinese Persona Chat (CPC) | 23,000 | Open |
13
  | LCCC | 11,987,759 | Open |
14
  | Emotional STC (ESTC) | 899,207 | Open |
15
+ | KdConv | 3,000 | Movie, Music, Travel |
16
+
17
+ # Example
18
+ ```python
19
+ #!/usr/bin/env python3
20
+ from transformers import BertTokenizer, BartForConditionalGeneration
21
+
22
+ tokenizer = BertTokenizer.from_pretrained("HIT-TMG/dialogue-bart-large-chinese")
23
+ model = BartForConditionalGeneration.from_pretrained("HIT-TMG/dialogue-bart-large-chinese")
24
+ '''