ratsgo/nlpbook

커스텀 데이터_분류_레이블별 성능체크 방법

Opened this issue · 0 comments

고수준으로 추상화해주셔서 작업을 완료 했는데
레이블별로 f-1 score나 성능을 확인해보고자 했지만 방법을 몰라서 여쭤봅니다
특히 데이터로드? ClassificationDataset? 레이블을 정수로 바꾸는거 같은데 원래 레이블과 맵핑하는 방법을 알려주시면 감사하겠습니다