pykeen/pykeen
🤖 A Python library for learning and evaluating knowledge graph embeddings
What it solves
PyKEEN은 지식 그래프 임베딩 모델의 학습 및 평가 과정을 단순화합니다. 연구자와 개발자가 모델이나 데이터셋을 처음부터 구현하지 않고도 멀티모달 정보를 포함한 다양한 임베딩 기법을 실험할 수 있는 표준화된 프레임워크를 제공합니다.
How it works
PyKEEN은 고수준 pipeline 함수를 사용하여 모델과 데이터셋을 지정하면 빠르게 학습 및 평가를 시작할 수 있습니다. 라이브러리는 확장성을 고려해 설계되었으며, 서로 다른 모델 및 학습 루프(예: 확률적 로컬 클로즈드 월드 가정) 간에 일관된 API를 제공하고, TriplesFactory와 같은 도구를 통해 사용자 정의 데이터셋을 손쉽게 다룰 수 있게 합니다.
Who it’s for
지식 그래프와 임베딩 모델을 다루는 AI 연구자 및 개발자를 위한 것으로, 모델 학습 및 평가를 위한 견고하고 확장 가능한 Python 패키지가 필요한 경우에 적합합니다.
Highlights
- 37개의 내장 데이터셋과 5개의 귀납 데이터셋을 포함합니다.
- 40가지 이상의 다양한 임베딩 모델을 지원합니다.
- Optuna와 통합하여 하이퍼파라미터 최적화를 수행하고, PyTorch Lightning과 통합해 확장 가능한 학습을 제공합니다.
- 고수준 pipeline을 제공하여 빠른 프로토타이핑과 평가가 가능합니다.