YanshekWoo
commited on
Commit
•
ab93ca6
1
Parent(s):
824e6bf
Update README.md
Browse files
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 |
+
'''
|