Commit
·
0484d16
1
Parent(s):
bd03818
Update README.md
Browse files
README.md
CHANGED
|
@@ -44,7 +44,8 @@ options = ["LOC", "PER", "ORG", "MISC"]
|
|
| 44 |
instruction = "please extract entities and their types from the input sentence, " \
|
| 45 |
"all entity types are in options"
|
| 46 |
|
| 47 |
-
text = "
|
|
|
|
| 48 |
|
| 49 |
generation_kwargs = {
|
| 50 |
"num_beams": 2,
|
|
@@ -58,5 +59,10 @@ pred_spans = model.predict(
|
|
| 58 |
options=options
|
| 59 |
)
|
| 60 |
|
| 61 |
-
>>>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
```
|
|
|
|
| 44 |
instruction = "please extract entities and their types from the input sentence, " \
|
| 45 |
"all entity types are in options"
|
| 46 |
|
| 47 |
+
text = "Once I visited Sovok in Nizny Novgorod. I had asian wok there. It was the best WOK i ever had"\
|
| 48 |
+
"It was cheap but lemonades cost 5 dollars."
|
| 49 |
|
| 50 |
generation_kwargs = {
|
| 51 |
"num_beams": 2,
|
|
|
|
| 59 |
options=options
|
| 60 |
)
|
| 61 |
|
| 62 |
+
>>> ('sovok is a Restaurant_Name, Nizny Novgorod is a Location, asian wok is a Dish, cheap is a Price, lemonades is a Dish, 5 dollars is a Price.',
|
| 63 |
+
[(24, 38, 'Location'),
|
| 64 |
+
(46, 55, 'Dish'),
|
| 65 |
+
(100, 105, 'Price'),
|
| 66 |
+
(110, 119, 'Dish'),
|
| 67 |
+
(125, 134, 'Price')])
|
| 68 |
```
|