Accenture/AmpliGraph

해결하는 문제

AmpliGraph는 지식 그래프 내 개념 간의 보이지 않거나 누락된 링크를 예측하도록 설계되었습니다. 사용자가 새로운 지식을 발견하고, 누락된 문장으로 대규모 지식 그래프를 완성하며, 해당 그래프 내 엔티티의 임베딩을 생성하는 것을 돕습니다.

작동 방식

이 라이브러리는 뉴럴 머신러닝 모델을 사용하여 지식 그래프 임베딩(메트릭 공간에서의 개념의 벡터 표현)을 생성합니다. 그런 다음 이러한 임베딩에 모델별 스코어링 함수를 적용하여 엔티티 간의 새로운 링크를 예측합니다.

대상 사용자

지식 그래프에서 관계 학습을 수행하거나, 새로운 관계 모델을 평가하거나, 엔티티 클러스터링 및 중복 예측과 같은 지식 발견 작업을 수행해야 하는 개발자와 연구자를 대상으로 합니다.

주요 특징

• Keras 스타일 API: 지식 그래프의 링크를 예측하는 모델 학습 프로세스를 단순화합니다.
• GPU 지원: TensorFlow 2를 기반으로 구축되어 CPU 및 GPU 장치 모두에서 원활한 학습 가속화가 가능합니다.
• 확장성: 사용자는 기본 추정기를 확장하여 자신만의 지식 그래프 임베딩 모델을 만들 수 있습니다.
• 다양한 모델 지원: TransE, DistMult, ComplEx, HolE, RotatE와 같은 모델의 구현이 포함되어 있습니다.