/
clewexx
/
hh
Обзор
Документация
Войти
/
clewexx
/
hh
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
utils/eval.py
18 строк
523 B
XingangPan
IBN-Net
26 июл 2018, 18:11
26 июл 2018, 18:11
8e45f35
Код
Авторство
О чём код?
from __future__ import print_function, absolute_import __all__ = ['accuracy'] def accuracy(output, target, topk=(1,)): """Computes the precision@k for the specified values of k""" maxk = max(topk) batch_size = target.size(0) _, pred = output.topk(maxk, 1, True, True) pred = pred.t() correct = pred.eq(target.view(1, -1).expand_as(pred)) res = [] for k in topk: correct_k = correct[:k].view(-1).float().sum(0) res.append(correct_k.mul_(100.0 / batch_size)) return res