1fc1755e7e28e0be952d722edac74f5d26cd78c8,opennmt/models/sequence_tagger.py,,flag_bioes_tags,#Any#Any#Any#,177

Before Change

      gold_tag = gold[b][index]
      pred_tag = predicted[b][index]

      if in_chunk:
        if pred_tag != gold_tag:
          _add_false_negative()
          index = _skip_chunk(gold[b], index)
          in_chunk = False
        elif gold_tag.startswith(b"E"):
          _add_true_positive()
          in_chunk = False
      else:
        if pred_tag == gold_tag:
          if gold_tag == b"O":
            _add_true_negative()
          elif gold_tag.startswith(b"S"):
            _add_true_positive()
          else:
            in_chunk = True
        else:
          if gold_tag == b"O":
            _add_false_positive()
          else:
            _add_false_negative()
            index = _skip_chunk(gold[b], index)

      index += 1

  return np.array(gold_flags), np.array(predicted_flags)

After Change

    while index < length:
      pred_tag = predicted[b][index]
      match, index = _match(predicted[b], gold[b], index, length)
      if not match and pred_tag != b"O":
        _add_false_positive()
      index += 1

Instances