Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,8 @@ You can use the model like this:
|
|
| 32 |
>>> 'Add To Playlist', 'Book Restaurant', 'Get Weather', 'Play Music', 'Rate Book', 'Search Creative Work',
|
| 33 |
>>> 'Search Screening Event'
|
| 34 |
>>> ]
|
| 35 |
-
>>>
|
|
|
|
| 36 |
>>> logits = model.predict(query, apply_softmax=True)
|
| 37 |
>>> print(logits)
|
| 38 |
|
|
|
|
| 32 |
>>> 'Add To Playlist', 'Book Restaurant', 'Get Weather', 'Play Music', 'Rate Book', 'Search Creative Work',
|
| 33 |
>>> 'Search Screening Event'
|
| 34 |
>>> ]
|
| 35 |
+
>>> aspect = 'intent'
|
| 36 |
+
>>> query = [[text, f'{lb} {aspect}'] for lb in labels]
|
| 37 |
>>> logits = model.predict(query, apply_softmax=True)
|
| 38 |
>>> print(logits)
|
| 39 |
|