fbpx
  • TorchKGE: библиотека для эмбеддинга графа знаний на PyTorch

    TorchKGE — это модуль на Python для эмбеддинга графа знаний. TorchKGE основан на Pytorch. Библиотека предоставляет исследователям и разработчикам API для разработки и тестирования новых моделей. TorchKGE содержит в себе структуру данных графа знаний, интерфейсы моделей и модули для негативного семплинга и оценки моделей.

    Подробнее про библиотеку

    Основным преимуществом библиотеки является модуль для оценки моделей предсказания связей. Задача предсказания связей является ключевой в применении к графу знаний. TorchKGE в 5 раз быстрее, чем AmpliGraph и в 24 раза быстрее OpenKE. Кроме того, библиотека содержит имплементации моделей для эмбеддинга графа знаний. 

    Обучение с ignite

    TorchKGE может использоваться совместно с PyTorch библиотекой ignite. Это упрощает использование ранней остановки (early stopping) во время обучения модели. Ниже — пример обучения TransE модели на датасете FB15k на GPU с ранней остановкой. Метрика оценки качества модели — MRR.

    Пример обучения модели в TorchKGE с библиотекой ignite