Python LogSoftmax Exemples

Exemple #1

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

Exemple #2

Fichier : metrics.py Projet : yaoshuyin/trax

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