コード例 #1
0
ファイル: metrics.py プロジェクト: yaoshuyin/trax
def SequenceAccuracy(classifier=core.ArgMax()):
    """Returns a layer that computes mean sequence prediction accuracy."""
    return cb.Serial(classifier,
                     _Accuracy(),
                     _WeightedSequenceMean(),
                     name='SequenceAccuracy',
                     sublayers_to_print=[])
コード例 #2
0
ファイル: metrics.py プロジェクト: yaoshuyin/trax
def Accuracy(classifier=core.ArgMax()):
    """Returns a layer that computes mean category prediction accuracy."""
    return cb.Serial(classifier,
                     _Accuracy(),
                     _WeightedMean(),
                     name='Accuracy',
                     sublayers_to_print=[])
コード例 #3
0
def SequenceAccuracy(classifier=core.ArgMax()):
    """Returns a layer that computes mean sequence prediction accuracy.

  DEPRECATED; use ``MaskedSequenceAccuracy`` instead.

  Args:
    classifier: Layer that transforms activation vectors into category
        predictions.
  """
    return cb.Serial(classifier,
                     _Accuracy(),
                     _WeightedSequenceMean(),
                     name='SequenceAccuracy',
                     sublayers_to_print=[])