datatypes
- class marius.tools.configuration.datatypes.InitOptions
- Return type
None
- class marius.tools.configuration.datatypes.UniformInitOptions(scale_factor=1)
- Parameters
scale_factor (float) –
- Return type
None
- scale_factor: float = 1
- class marius.tools.configuration.datatypes.NormalInitOptions(mean=0, std=1)
- Parameters
mean (float) –
std (float) –
- Return type
None
- mean: float = 0
- std: float = 1
- class marius.tools.configuration.datatypes.ConstantInitOptions(constant=0)
- Parameters
constant (float) –
- Return type
None
- constant: float = 0
- class marius.tools.configuration.datatypes.LossOptions(reduction='SUM')
- Parameters
reduction (str) –
- Return type
None
- reduction: str = 'SUM'
- class marius.tools.configuration.datatypes.RankingLossOptions(reduction='SUM', margin=0.1)
- Parameters
reduction (str) –
margin (float) –
- Return type
None
- margin: float = 0.1
- class marius.tools.configuration.datatypes.OptimizerOptions(learning_rate=0.1)
- Parameters
learning_rate (float) –
- Return type
None
- learning_rate: float = 0.1
- class marius.tools.configuration.datatypes.AdagradOptions(learning_rate=0.1, eps=1e-10, init_value=0, lr_decay=0, weight_decay=0)
- Parameters
learning_rate (float) –
eps (float) –
init_value (float) –
lr_decay (float) –
weight_decay (float) –
- Return type
None
- learning_rate: float = 0.1
- eps: float = 1e-10
- init_value: float = 0
- lr_decay: float = 0
- weight_decay: float = 0
- class marius.tools.configuration.datatypes.AdamOptions(learning_rate=0.1, amsgrad=False, beta_1=0.9, beta_2=0.999, eps=1e-08, weight_decay=0)
- Parameters
learning_rate (float) –
amsgrad (bool) –
beta_1 (float) –
beta_2 (float) –
eps (float) –
weight_decay (float) –
- Return type
None
- learning_rate: float = 0.1
- amsgrad: bool = False
- beta_1: float = 0.9
- beta_2: float = 0.999
- eps: float = 1e-08
- weight_decay: float = 0
- class marius.tools.configuration.datatypes.LayerOptions
- Return type
None
- class marius.tools.configuration.datatypes.EmbeddingLayerOptions
- Return type
None
- class marius.tools.configuration.datatypes.FeatureLayerOptions
- Return type
None
- class marius.tools.configuration.datatypes.DenseLayerOptions(type='LINEAR')
- Parameters
type (str) –
- Return type
None
- type: str = 'LINEAR'
- class marius.tools.configuration.datatypes.ReductionLayerOptions(type='CONCAT')
- Parameters
type (str) –
- Return type
None
- type: str = 'CONCAT'
- class marius.tools.configuration.datatypes.GNNLayerOptions(type)
- Parameters
type (str) –
- Return type
None
- type: str
- class marius.tools.configuration.datatypes.GraphSageLayerOptions(type='GRAPH_SAGE', aggregator='GCN')
- Parameters
type (str) –
aggregator (str) –
- Return type
None
- type: str = 'GRAPH_SAGE'
- aggregator: str = 'GCN'
- class marius.tools.configuration.datatypes.GATLayerOptions(type='GAT', num_heads=10, average_heads=True, negative_slope=0.2, input_dropout=0.0, attention_dropout=0.0)
- Parameters
type (str) –
num_heads (int) –
average_heads (bool) –
negative_slope (float) –
input_dropout (float) –
attention_dropout (float) –
- Return type
None
- type: str = 'GAT'
- num_heads: int = 10
- average_heads: bool = True
- negative_slope: float = 0.2
- input_dropout: float = 0.0
- attention_dropout: float = 0.0
- class marius.tools.configuration.datatypes.DecoderOptions
- Return type
None
- class marius.tools.configuration.datatypes.EdgeDecoderOptions(inverse_edges=True, use_relation_features=False, edge_decoder_method='CORRUPT_NODE')
- Parameters
inverse_edges (bool) –
use_relation_features (bool) –
edge_decoder_method (str) –
- Return type
None
- inverse_edges: bool = True
- use_relation_features: bool = False
- edge_decoder_method: str = 'CORRUPT_NODE'
- class marius.tools.configuration.datatypes.StorageOptions(dtype='float')
- Parameters
dtype (str) –
- Return type
None
- dtype: str = 'float'
- class marius.tools.configuration.datatypes.PartitionBufferOptions(dtype='float', num_partitions=16, buffer_capacity=8, prefetching=True, fine_to_coarse_ratio=1, num_cache_partitions=0, edge_bucket_ordering='NEW_BETA', node_partition_ordering='DISPERSED', randomly_assign_edge_buckets=True)
- Parameters
dtype (str) –
num_partitions (int) –
buffer_capacity (int) –
prefetching (bool) –
fine_to_coarse_ratio (int) –
num_cache_partitions (int) –
edge_bucket_ordering (str) –
node_partition_ordering (str) –
randomly_assign_edge_buckets (bool) –
- Return type
None
- num_partitions: int = 16
- buffer_capacity: int = 8
- prefetching: bool = True
- fine_to_coarse_ratio: int = 1
- num_cache_partitions: int = 0
- edge_bucket_ordering: str = 'NEW_BETA'
- node_partition_ordering: str = 'DISPERSED'
- randomly_assign_edge_buckets: bool = True
- class marius.tools.configuration.datatypes.NeighborSamplingOptions
- Return type
None