Exemplo n.º 1
0
def get_pred_true(h_params: HParams) -> Iterable[NERData]:
    """Get true values for prediction."""
    lines = get_ner_lines(Path(h_params.data_dir) / 'test.txt')

    def helper(item: List[str]) -> NERData:
        sentence = item[1].split(' ')
        true = item[0].split(' ')
        pred = []
        return NERData(sentence, true, pred)

    return map(helper, lines)
Exemplo n.º 2
0
def test_get_ner_lines():
    lines = get_ner_lines(data_dir / 'test.txt')
    assert len(lines) == 105
    assert lines[0][0] == 'C O O O O B-StationDest'
    assert lines[0][1] == 'FindConnection i want to go marienplatz'
Exemplo n.º 3
0
 def get_test_examples(self, data_dir):
     return self._create_example(get_ner_lines(Path(data_dir) / 'test.txt'),
                                 "test")
Exemplo n.º 4
0
 def get_dev_examples(self, data_dir):
     tf.logging.warning('Returned dev examples are from test set.')
     return self._create_example(get_ner_lines(Path(data_dir) / 'test.txt'),
                                 "dev")
Exemplo n.º 5
0
 def get_train_examples(self, data_dir):
     return self._create_example(
         get_ner_lines(Path(data_dir) / 'train.txt'), "train")
Exemplo n.º 6
0
 def _read_data(cls, input_file):
     """Reads a BIO data."""
     from improv.read_ner import get_ner_lines
     return get_ner_lines(Path(input_file))