Python pyrouge 패키지 설치

 

Summarization task를 돌리다 보면 pyrouge 패키지를 이용하는 것을 볼 수 있다.

그런데, 단순히 pip install pyrouge 하면 끝나는 게 아니라, 추가적인 설정이 필요하다.

일단, pyrouge 프로젝트의 github의 README를 보면 다음과 같다.

'pip install pyrouge'를 실행하면 여러 binary 실행 파일도 같이 설치되며, 그 중 하나가 pyrouge_set_rouge_path'이다.

인자로 '/absolute/path/to/ROUGE-1.5.5/directory'를 설정해야 하는데, 여기서 말하는 ROUGE-1.5.5는 https://github.com/andersjo/pyrouge의 './tools/ROUGE-1.5.5/'를 의미한다.

프로젝트를 다운받고, 경로를 설정해 준다.

그러면, 'python -m pyrouge.test'가 성공하는 것처럼 나오는데, 실제 실행할 경우 'Cannot open exception db file for reading: ~/Projects_PATH/rouge/tools/ROUGE-1.5.5/data/WordNet-2.0.exc.db' 이런 메세지가 나왔다.

한동안 해결을 못하고 있다가, 다른 Summarization repo를 보다가 해결책이 보였다.

아마도 이미 있는 파일의 버전이 잘못되어 있는 것 같다.

위의 명령어를 실행해 WordNet-2.0.exc.db 파일을 다시 만들어주면 test가 완료된다.