task_categories:translation task_categories:token-classification license:mit doi:10.57967/hf/0871 region:us
---
license: mit
task_categories:
- translation
- token-classification
---


# Natural Language to State Translation Dataset<br>

This dataset can be used for translation or token classification tasks. There are two versions: 'nl-st' contains over 1.2 million records and 'nl-st-lg' contains over 9.8 million records. Each record has 6 features:<br>
- ***sentence*** (string) - natural language (English) sentence that describes the state.
- ***state*** (string) - state information consisting of percept value pairs stored as a string (percept value)
- ***ner_tags*** (string[]) - NER tags for each token in the sentence stored as a list
- ***ner_sentence*** (string) - string of NER tags concatenated together representing the sentence
- ***ner_ids*** (float[]) - the ids corresponding to the ner_tags
- ***tokens*** (string[]) - sentence split into each token the corresponds to an ner_ids and ner_tags
<br>

## Maps
### NER ID to NER Tag Map<br>
```
ner_id_map = {
    0: "O",
    1: "L-DEMO",
    2: "L-BA",
    3: "V-BA",
    4: "L-GROUND",
    5: "L-BALL",
    6: "L-SPEED",
    7: "V-SPEED",
    8: "L-DIR",
    9: "V-DIR",
    10: "L-BRAKE",
    11: "L-STEER",
    12: "V-STEER",
    13: "L-THROTTLE",
    14: "V-THROTTLE",
    15: "L-BOOST",
    16: "L-POS"
}
```
### NER Tag to NER ID Map
```
ner_tag_map = {
    "O": 0,
    "L-DEMO": 1,
    "L-BA": 2,
    "V-BA": 3,
    "L-GROUND": 4,
    "L-BALL": 5,
    "L-SPEED": 6,
    "V-SPEED": 7,
    "L-DIR": 8,
    "V-DIR": 9,
    "L-BRAKE": 10,
    "L-STEER": 11,
    "V-STEER": 12,
    "L-THROTTLE": 13,
    "V-THROTTLE": 14,
    "L-BOOST": 15,
    "L-POS": 16
}

```