pykeen/pykeen

해결하는 문제

PyKEEN은 지식 그래프 임베딩 모델을 훈련하고 평가하는 과정을 간소화합니다. 지식 그래프 데이터, 모델 구현, 성능 평가의 복잡성을 표준화된 프레임워크로 처리하며, 다중 모달 정보를 지원합니다.

작동 방식

PyKEEN은 고수준의 pipeline 함수를 제공하여, 최소한의 코드로 특정 데이터셋(예: Nations)에서 모델(예: TransE)을 빠르게 훈련하고 평가할 수 있습니다. PyTorch Lightning을 기반으로 하며, 하이퍼파라미터 최적화를 위해 Optuna와 통합됩니다. 이 라이브러리는 확장성을 고려하여 모델, 훈련 루프(예: 확률적 국소 폐쇄 세계 가정), 사용자 정의 데이터 로딩을 위한 삼중항 공장(TriplesFactory)에 일관된 API를 제공합니다.

대상 사용자

지식 그래프를 다루는 연구자 및 개발자 중, 엔티티와 관계를 임베딩하여 링크 예측과 같은 작업을 수행하기 위해 강력한 도구가 필요한 사람들을 대상으로 합니다.

주요 특징

• 37개의 내장 데이터셋과 5개의 유도형 데이터셋 포함.
• 40개 이상의 다양한 임베딩 모델 구현.
• 모델과 훈련 루프를 통합된 API로 교체 가능.
• TriplesFactory를 통해 사용자 정의 데이터셋 통합 지원.