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 |
|