Accenture/AmpliGraph

解決する課題

AmpliGraphは、知識グラフ内の概念間における未知または欠損しているリンクを予測するために設計されています。ユーザーが新しい知識を発見し、欠落している記述で大規模な知識グラフを補完し、グラフ内のエンティティの埋め込みを生成するのを支援します。

仕組み

このライブラリは、ニューラル機械学習モデルを使用して、知識グラフの埋め込み（メトリック空間における概念のベクトル表現）を作成します。その後、これらの埋め込みにモデル固有のスコアリング関数を適用して、エンティティ間の新しいリンクを予測します。

対象ユーザー

知識グラフ上でのリレーショナル学習の実行、新しいリレーショナルモデルの評価、またはエンティティのクラスタリングや重複予測などの知識発見タスクを必要とする開発者や研究者向けに設計されています。

特徴

• KerasスタイルのAPI: 知識グラフのリンクを予測する学習モデルのプロセスを簡素化します。
• GPU対応: TensorFlow 2に基づいて構築されており、CPUとGPUの両方のデバイスでシームレスなトレーニング加速が可能です。
• 拡張性: ユーザーはベースとなるエスティメータを拡張することで、独自の知識グラフ埋め込みモデルを作成できます。
• 複数のモデルをサポート: TransE、DistMult、ComplEx、HolE、RotatEなどのモデルの実装が含まれています。