Update README.md
Browse files
README.md
CHANGED
|
@@ -42,8 +42,10 @@ You can use the model like this:
|
|
| 42 |
>>> 'Search Screening Event'
|
| 43 |
>>> ]
|
| 44 |
>>> aspect = 'intent'
|
|
|
|
|
|
|
| 45 |
|
| 46 |
-
>>> query = [[text,
|
| 47 |
>>> logits = model.predict(query, apply_softmax=True)
|
| 48 |
>>> print(logits)
|
| 49 |
|
|
|
|
| 42 |
>>> 'Search Screening Event'
|
| 43 |
>>> ]
|
| 44 |
>>> aspect = 'intent'
|
| 45 |
+
>>> sep_token = '<|ASPECT-SEP|>'
|
| 46 |
+
>>> text = f'{aspect} {sep_token} {text}'
|
| 47 |
|
| 48 |
+
>>> query = [[text, lb] for lb in labels]
|
| 49 |
>>> logits = model.predict(query, apply_softmax=True)
|
| 50 |
>>> print(logits)
|
| 51 |
|