smrunner.metrics Namespace Reference

Classes
class	AverageMeter
class	AverageMeterDict

Functions
	top1_acc (data, pred_key, gt_key, target_idx=None)
	top1_acc_dict (data, pred_key, gt_key)
	top1_acc_multi_task (data, pred_key, gt_key, target_tasks, task_key)

Function Documentation

top1_acc()

smrunner.metrics.top1_acc	(		data,
			pred_key,
			gt_key,
			target_idx = None )

top1 정확도를 계산하는 기능
args:
    data ():
    pred_key ():
    target_idx ():

Definition at line 91 of file metrics.py.

def top1_acc(data, pred_key, gt_key, target_idx=None):
    """
    top1 정확도를 계산하는 기능
    args:
        data ():
        pred_key ():
        target_idx ():
    """
    if target_idx is None:
        return data[pred_key] == data[gt_key]
    else:
        return data[pred_key][target_idx] == data[gt_key][target_idx]
 

top1_acc_dict()

smrunner.metrics.top1_acc_dict	(		data,
			pred_key,
			gt_key )

여러 결과에 대한 top1 정확도를 계산하는 기능

Definition at line 104 of file metrics.py.

def top1_acc_dict(data, pred_key, gt_key):
    """
    여러 결과에 대한 top1 정확도를 계산하는 기능
    """
    acc = {}
    for p_key, g_key in zip(pred_key, gt_key):
        acc[p_key] = top1_acc(data, p_key, g_key)
 
    return acc
 

top1_acc_multi_task()

smrunner.metrics.top1_acc_multi_task	(		data,
			pred_key,
			gt_key,
			target_tasks,
			task_key )

특정 테스크에 대한 top1 정확도를 계산하는 기능

Definition at line 114 of file metrics.py.

def top1_acc_multi_task(data, pred_key, gt_key, target_tasks, task_key):
    """
    특정 테스크에 대한 top1 정확도를 계산하는 기능
    """
    acc = {}
    for p_key, g_key, t_task in zip(pred_key, gt_key, target_tasks):
        t_idx = [task == t_task for task in data[task_key]]
        t_idx = torch.tensor(t_idx)
 
        acc[p_key] = top1_acc(data, p_key, g_key, t_idx)
    return acc