GitHub / lorenzobalzani / neural-subgraph-retrieval
PyTorch module that generates a subgraph originating from the UMLS Knowledge Graph (KG) when presented with a biomedical question (from MedQA). It utilizes Language Models (LMs) to solve the next relation prediction task and graph node embedding to infer the tail given a head and the predicted relation, essentially performing link prediction.
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/lorenzobalzani%2Fneural-subgraph-retrieval
PURL: pkg:github/lorenzobalzani/neural-subgraph-retrieval
Stars: 0
Forks: 0
Open issues: 0
License: None
Language: Python
Size: 8.42 MB
Dependencies parsed at: Pending
Created at: over 1 year ago
Updated at: over 1 year ago
Pushed at: over 1 year ago
Last synced at: over 1 year ago
Topics: question-answering, subgraph-retrieval, umls