数据集:
tner/mit_restaurant
MIT Restaurant NER dataset formatted in a part of TNER project.
An example of train looks as follows.
{ 'tags': [0, 0, 0, 0, 0, 0, 0, 0, 5, 3, 4, 0], 'tokens': ['can', 'you', 'find', 'the', 'phone', 'number', 'for', 'the', 'closest', 'family', 'style', 'restaurant'] }
The label2id dictionary can be found at here .
{ "O": 0, "B-Rating": 1, "I-Rating": 2, "B-Amenity": 3, "I-Amenity": 4, "B-Location": 5, "I-Location": 6, "B-Restaurant_Name": 7, "I-Restaurant_Name": 8, "B-Price": 9, "B-Hours": 10, "I-Hours": 11, "B-Dish": 12, "I-Dish": 13, "B-Cuisine": 14, "I-Price": 15, "I-Cuisine": 16 }
name | train | validation | test |
---|---|---|---|
mit_restaurant | 6900 | 760 | 1521 |