Function classify_step

functions.py:156–161

(logit, classes)

def classify_step(logit, classes):
    class_vector = F.softmax(logit, 1).data.squeeze()
    assert len(class_vector) == len(classes), "class number must match"
    probs, idx = class_vector.sort(0, True)
    result = classes[idx[0]]
    return result