Python LogSoftmax Examples

Example #1

def CrossEntropyLossWithLogSoftmax():
    """Mean prediction-target cross-entropy for multiclass classification."""
    return cb.Serial(core.LogSoftmax(),
                     _CrossEntropy(),
                     _WeightedMean(),
                     name='CrossEntropyLossWithLogSoftmax',
                     sublayers_to_print=[])

Example #2

File: metrics.py Project: yaoshuyin/trax

def CrossEntropyLossWithLogSoftmax():
    """Mean prediction-target cross entropy for multiclass classification."""
    return cb.Serial(core.LogSoftmax(),
                     CrossEntropyLoss(),
                     name='CrossEntropyLoss')