5f9be68993f6f1809804ac373db9d6fb7f397d0f,opennmt/models/sequence_tagger.py,SequenceTagger,compute_metrics,#SequenceTagger#Any#Any#,116

Before Change

      recall_metric = tf.metrics.recall(gold_flags, predicted_flags)

      precision = precision_metric[0]
      recall = recall_metric[0]
      f1 = (2 * precision * recall) / (recall + precision)

      eval_metric_ops["precision"] = precision_metric
      eval_metric_ops["recall"] = recall_metric

After Change

          [labels["tags"], predictions["tags"], labels["length"]],
          [tf.bool, tf.bool])

      f1 = F1()
      f1.update_state(gold_flags, predicted_flags)
      eval_metric_ops["precision"] = f1.precision
      eval_metric_ops["recall"] = f1.recall
      eval_metric_ops["f1"] = f1

Instances