Log In Sign Up
Models:
Amanda Davis
Amanda-D/
torchdrug
0
Downloads: 1
[3eb847]: / doc / source / api / models.rst

Download this file

175 lines (128 with data), 2.4 kB

torchdrug.models

Knowledge Graph Reasoning Models

TransE

DistMult

ComplEx

SimplE

RotatE

NeuralLP

KBGAT

Graph Neural Networks

ChebNet

GCN

GAT

GIN

MPNN

NFP

RGCN

GearNet

SchNet

Protein Sequence Encoders

ESM

ProteinCNN

ProteinResNet

ProteinLSTM

ProteinBERT

Statistic Feature Engineering

Physicochemical Feature Engineering

Normalizing Flows

GraphAutoregressiveFlow

Self-supervised Models

InfoGraph

MultiviewContrast