pykeen: 지식 그래프 임베딩 모델의 학습 및 평가를 위한 Python 프레임워크

해결하는 문제

PyKEEN은 지식 그래프 임베딩 모델의 학습 및 평가 프로세스를 단순화합니다. 연구자와 개발자가 모델이나 데이터셋을 처음부터 직접 구현할 필요 없이, 멀티모달 정보를 포함하는 기술을 포함하여 다양한 임베딩 기술을 실험할 수 있는 표준화된 프레임워크를 제공합니다.

작동 방식

PyKEEN은 사용자가 모델과 데이터셋을 지정하여 학습 및 평가를 빠르게 시작할 수 있도록 하는 고수준 pipeline 함수를 사용합니다. 이 라이브러리는 확장 가능하도록 설계되었으며, 다양한 모델과 학습 루프(예: stochastic local closed world assumption)에 걸쳐 일관된 API를 제공하고, 사용자 정의 데이터셋을 처리하기 위한 TriplesFactory와 같은 도구를 제공합니다.

대상 사용자

지식 그래프 및 임베딩 모델을 다루며 모델 학습 및 평가를 위한 견고하고 확장 가능한 Python 패키지가 필요한 AI 연구자 및 개발자를 위해 설계되었습니다.

주요 특징

• 37개의 내장 데이터셋과 5개의 귀납적(inductive) 데이터셋을 포함합니다.
• 40개의 서로 다른 임베딩 모델을 지원합니다.
• 하이퍼파라미터 최적화를 위한 Optuna 및 확장 가능한 학습을 위한 PyTorch Lightning과 통합되어 있습니다.
• 신속한 프로토타이핑 및 평가를 위한 고수준 파이프라인을 제공합니다.