uer commited on
Commit
fde00b5
·
1 Parent(s): 4c11bab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -15
README.md CHANGED
@@ -50,26 +50,26 @@ You can use this model directly with a pipeline for masked language modeling (ta
50
  >>> unmasker("中国的首都是[MASK]京。")
51
  [
52
  {'sequence': '[CLS] 中 国 的 首 都 是 北 京 。 [SEP]',
53
- 'score': 0.8722995519638062,
54
- 'token': 1266,
55
- 'token_str': '北'},
56
- {'sequence': '[CLS] 中 国 的 首 都 是 南 京 。 [SEP]',
57
- 'score': 0.06813988834619522,
58
  'token': 1298,
59
  'token_str': '南'},
60
  {'sequence': '[CLS] 中 国 的 首 都 是 东 京 。 [SEP]',
61
- 'score': 0.05202966928482056,
62
- 'token': 691,
63
- 'token_str': '东'},
64
  {'sequence': '[CLS] 中 国 的 首 都 是 普 京 。 [SEP]',
65
- 'score': 0.0035027158446609974,
66
  'token': 3249,
67
- 'token_str': '普'},
68
- {'sequence': '[CLS] 中 国 的 首 都 是 京 。 [SEP]',
69
- 'score': 0.0005988680641166866,
70
- 'token': 1426,
71
- 'token_str': ''}
72
- ]
73
  ```
74
 
75
  Here is how to use this model to get the features of a given text in PyTorch:
 
50
  >>> unmasker("中国的首都是[MASK]京。")
51
  [
52
  {'sequence': '[CLS] 中 国 的 首 都 是 北 京 。 [SEP]',
53
+ 'score': 0.9338967204093933,
54
+ 'token': 1266,
55
+ 'token_str': '北'},
56
+ {'sequence': '[CLS] 中 国 的 首 都 是 南 京 。 [SEP]',
57
+ 'score': 0.039428312331438065,
58
  'token': 1298,
59
  'token_str': '南'},
60
  {'sequence': '[CLS] 中 国 的 首 都 是 东 京 。 [SEP]',
61
+ 'score': 0.01681734062731266,
62
+ 'token': 691,
63
+ 'token_str': '东'},
64
  {'sequence': '[CLS] 中 国 的 首 都 是 普 京 。 [SEP]',
65
+ 'score': 0.004590896889567375,
66
  'token': 3249,
67
+ 'token_str': '普'},
68
+ {'sequence': '[CLS] 中 国 的 首 都 是 京 。 [SEP]',
69
+ 'score': 0.0007656012894585729,
70
+ 'token': 4242,
71
+ 'token_str': ''}
72
+ ]
73
  ```
74
 
75
  Here is how to use this model to get the features of a given text in PyTorch: