search

seujung / KoBART-summarization

Summarization module based on KoBART
MIT License
196 stars 89 forks source link

퍼포먼스 측정에 대해 여쭤보고 싶습니다. #22

Open edhz8 opened 2 years ago

edhz8 commented 2 years ago

Score 산출 방법은 Dacon 한국어 문서 생성요약 AI 경진대회 metric을 활용함 이라고 남겨놓으셨던데 해당부분이 rouge_metric.py 맞나요? 이 파일 사용방법이 궁금합니다.

seujung commented 2 years ago

안녕하세요. 사용법을 추가해서 업로드 하도록 하겠습니다.

soom1017 commented 2 years ago

https://dacon.io/competitions/official/235673/talkboard/401911?page=1&dtype=recent 이 곳의 RougeScorer 코드 복사해서 class 정의하고 아래와 같이 작성하시면 될 것 같습니다.

scorer = RougeScorer()
scorer.compute_rouge(df_test, df_output) # df_test: test.tsv의 Dataframe, df_output: 같은 형식의 Dataframe