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 を提供し、迅速なプロトタイピングと評価が可能。