
[PyTorch] MultilabelAUPRC
·
코딩/이슈
`pytorch`를 사용하면 메트릭을 계산할 때에 저는 주로 `torchmetrics`를 사용합니다. 실험 조건이나 데이터의 상태에 따라 여러 메트릭이 필요한데, `torchmetrics`에는 부족함 없이 구현되어있기 때문입니다. 그런데 신기하게도 멀티라벨 에서의 `AUC-PR`은 없습니다. 저 메트릭을 위해 조사하다 `torcheval`이라는 라이브러리를 찾았습니다. 기본적으로 `torchmetrics`과 매우 유사하지만, `AUC-PR`이 구현되어 있습니다. 함수 불러오는것 까지는 `torchmetrics`과 똑같은데, 함수형으로의 구현은 안되어있는듯 합니다. 멀티라벨 이다보니 기본적으로 prediction & real target 모두 벡터가 아닌 원핫인코딩 된 행렬로 넣으면 됩니다.>>> impo..