pykeen/pykeen

何を解決するか

PyKEENは知識グラフ埋め込みモデルのトレーニングと評価プロセスを簡素化します。知識グラフデータ、モデル実装、パフォーマンス評価の複雑さを標準化されたフレームワークで扱い、マルチモーダル情報のサポートも提供します。

どう動くか

PyKEENは高レベルの pipeline 関数を提供し、最小限のコードで特定のデータセット（例：Nations）上でモデル（例：TransE）を迅速にトレーニングおよび評価できます。PyTorch Lightning を基盤とし、ハイパーパラメータ最適化には Optuna を統合しています。このライブラリは拡張性を重視しており、モデル、トレーニングループ（例：確率的局所閉世界仮説）、カスタムデータインジェスト用の三つ組工場（TriplesFactory）に対して一貫したAPIを提供します。

対象ユーザー

知識グラフを扱う研究者や開発者で、エンティティと関係の埋め込みを実行し、リンク予測などのタスクを遂行するための堅牢なツールが必要な人向けです。

主な特徴

• 37の組み込みデータセットと5つの誘導型データセットを含む。
• 40種類の異なる埋め込みモデルを実装。
• モデルとトレーニングループの切り替えを統一されたAPIで可能。
• TriplesFactory を通じたカスタムデータセットの統合をサポート。