Struct SiameseGraphNetwork 

pub struct SiameseGraphNetwork { /* private fields */ }

Siamese Graph Network for similarity learning

Implementations

impl SiameseGraphNetwork

pub fn new( input_dim: usize, hidden_dim: usize, output_dim: usize, ) -> Result<Self, TorshError>

Create a new Siamese graph network

pub fn embed(&self, graph: &GraphData) -> Result<Tensor, TorshError>

Compute embeddings for a graph

pub fn contrastive_loss( &self, graph1: &GraphData, graph2: &GraphData, is_similar: bool, margin: f32, ) -> Result<f32, TorshError>

Compute contrastive loss between similar and dissimilar pairs

Trait Implementations

impl Debug for SiameseGraphNetwork

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

Formats the value using the given formatter.