IRIS-AUDIO / SELD

7 stars 8 forks source link

focal loss 추가 + 클래스별 리콜와 프리시전 #109

Closed ParkJinHyeock closed 3 years ago

ParkJinHyeock commented 3 years ago

각 class별로 recall과 precision을 계산해서 텐서보드에 저장하는 것과 loss를 focal loss와 bce중에 선택하는 코드를 추가하였습니다. config를 이용해 알파와 gamma를 조절할 수 있습니다.

lsg1213 commented 3 years ago

mixing은 뭔지 알 수 있을까요?