pub struct GNNConfig {
pub num_layers: usize,
pub hidden_dim: usize,
pub output_dim: usize,
pub aggregation: AggregationType,
pub activation: ActivationType,
pub dropout: f64,
pub add_self_loops: bool,
pub layer_norm: bool,
}Expand description
Configuration for GNN inference.
Fields§
§num_layers: usizeNumber of message passing layers.
Hidden dimension size.
output_dim: usizeOutput dimension (number of classes or embedding size).
aggregation: AggregationTypeAggregation function for messages.
activation: ActivationTypeActivation function.
dropout: f64Dropout rate (0-1).
add_self_loops: boolWhether to add self-loops.
layer_norm: boolWhether to use layer normalization.
Trait Implementations§
Source§impl<'de> Deserialize<'de> for GNNConfig
impl<'de> Deserialize<'de> for GNNConfig
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for GNNConfig
impl RefUnwindSafe for GNNConfig
impl Send for GNNConfig
impl Sync for GNNConfig
impl Unpin for GNNConfig
impl UnwindSafe for GNNConfig
Blanket Implementations§
Source§impl<T> ArchivePointee for T
impl<T> ArchivePointee for T
Source§type ArchivedMetadata = ()
type ArchivedMetadata = ()
The archived version of the pointer metadata for this type.
Source§fn pointer_metadata(
_: &<T as ArchivePointee>::ArchivedMetadata,
) -> <T as Pointee>::Metadata
fn pointer_metadata( _: &<T as ArchivePointee>::ArchivedMetadata, ) -> <T as Pointee>::Metadata
Converts some archived metadata to the pointer metadata for itself.
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<F, W, T, D> Deserialize<With<T, W>, D> for F
impl<F, W, T, D> Deserialize<With<T, W>, D> for F
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> LayoutRaw for T
impl<T> LayoutRaw for T
Source§fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
fn layout_raw(_: <T as Pointee>::Metadata) -> Result<Layout, LayoutError>
Gets the layout of the type.