예제 #1
0
def _get_evaluator(metric):
    if metric == "accuracy":
        return Accuracy()
    elif metric == "multilabel_f1":
        return MultiLabelMicroF1()
    elif metric == "multiclass_f1":
        return MultiClassMicroF1()
    else:
        raise NotImplementedError
예제 #2
0
파일: dataset.py 프로젝트: rpatil524/cogdl 
 def get_evaluator(self):
     return Accuracy()