TorchKGE — это модуль на Python для эмбеддинга графа знаний. TorchKGE основан на Pytorch. Библиотека предоставляет исследователям и разработчикам API для разработки и тестирования новых моделей. TorchKGE содержит в себе структуру данных графа знаний, интерфейсы моделей и модули для негативного семплинга и оценки моделей.
Подробнее про библиотеку
Основным преимуществом библиотеки является модуль для оценки моделей предсказания связей. Задача предсказания связей является ключевой в применении к графу знаний. TorchKGE в 5 раз быстрее, чем AmpliGraph и в 24 раза быстрее OpenKE. Кроме того, библиотека содержит имплементации моделей для эмбеддинга графа знаний.
Обучение с ignite
TorchKGE может использоваться совместно с PyTorch библиотекой ignite. Это упрощает использование ранней остановки (early stopping) во время обучения модели. Ниже — пример обучения TransE модели на датасете FB15k на GPU с ранней остановкой. Метрика оценки качества модели — MRR.
