Enum GNNType 

pub enum GNNType {
    GCN,
    GraphSAGE,
    GAT,
    GraphTransformer,
    GIN,
    PNA,
    HetGNN,
    TGN,
}

Type of GNN architecture

Variants

GCN

Graph Convolutional Network

GraphSAGE

GraphSAGE - Sampling and aggregating

GAT

Graph Attention Network

GraphTransformer

Graph Transformer

GIN

Graph Isomorphism Network

PNA

Principal Neighbourhood Aggregation

HetGNN

Heterogeneous Graph Network

TGN

Temporal Graph Network

Implementations

impl GNNType

pub fn default_layers(&self) -> usize

pub fn requires_attention(&self) -> bool

Trait Implementations

impl Clone for GNNType

fn clone(&self) -> GNNType

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GNNType

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for GNNType

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for GNNType

fn eq(&self, other: &GNNType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for GNNType

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for GNNType

impl StructuralPartialEq for GNNType