Index

_

__init__() (marius.data.DataLoader method), [1], [2] (marius.data.DENSEGraph method), [1] (marius.data.MariusGraph method), [1] (marius.data.samplers.CorruptNodeNegativeSampler method) (marius.data.samplers.LayeredNeighborSampler method), [1], [2] (marius.data.samplers.RandomEdgeSampler method) (marius.nn.AdagradOptimizer method), [1] (marius.nn.AdamOptimizer method), [1] (marius.nn.BCEAfterSigmoidLoss method) (marius.nn.BCEWithLogitsLoss method) (marius.nn.CrossEntropyLoss method) (marius.nn.decoders.Decoder method) (marius.nn.decoders.edge.Comparator method) (marius.nn.decoders.edge.ComplEx method) (marius.nn.decoders.edge.ComplexHadamardOperator method) (marius.nn.decoders.edge.CosineCompare method) (marius.nn.decoders.edge.DistMult method) (marius.nn.decoders.edge.DotCompare method) (marius.nn.decoders.edge.EdgeDecoder method) (marius.nn.decoders.edge.HadamardOperator method) (marius.nn.decoders.edge.L2Compare method) (marius.nn.decoders.edge.NoOp method) (marius.nn.decoders.edge.RelationOperator method) (marius.nn.decoders.edge.TransE method) (marius.nn.decoders.edge.TranslationOperator method) (marius.nn.decoders.node.NodeDecoder method) (marius.nn.decoders.node.NoOpNodeDecoder method) (marius.nn.encoders.GeneralEncoder method), [1] (marius.nn.layers.ConcatReduction method), [1] (marius.nn.layers.EmbeddingLayer method), [1] (marius.nn.layers.FeatureLayer method), [1] (marius.nn.layers.GATLayer method), [1] (marius.nn.layers.GCNLayer method), [1] (marius.nn.layers.GNNLayer method) (marius.nn.layers.GraphSageLayer method), [1] (marius.nn.layers.Layer method) (marius.nn.layers.LinearReduction method), [1] (marius.nn.layers.ReductionLayer method) (marius.nn.LossFunction method) (marius.nn.Model method), [1] (marius.nn.MSELoss method) (marius.nn.Optimizer method) (marius.nn.RankingLoss method) (marius.nn.SGDOptimizer method) (marius.nn.SoftmaxCrossEntropy method) (marius.nn.SoftPlusLoss method) (marius.pipeline.Evaluator method) (marius.pipeline.GraphEncoder method) (marius.pipeline.PipelineEncoder method) (marius.pipeline.PipelineEvaluator method) (marius.pipeline.PipelineTrainer method) (marius.pipeline.SynchronousEncoder method) (marius.pipeline.SynchronousEvaluator method) (marius.pipeline.SynchronousTrainer method) (marius.pipeline.Trainer method) (marius.report.LinkPredictionReporter method) (marius.report.NodeClassificationReporter method) (marius.report.ProgressReporter method) (marius.report.Reporter method) (marius.storage.FlatFile method), [1], [2] (marius.storage.GraphModelStorage method), [1] (marius.storage.InMemory method), [1], [2] (marius.storage.PartitionBufferStorage method), [1], [2] (marius.storage.Storage method) (marius.tools.preprocess.dataset.Dataset method) (marius.tools.preprocess.datasets.fb15k.FB15K method) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 method) (marius.tools.preprocess.datasets.freebase86m.Freebase86m method) (marius.tools.preprocess.datasets.friendster.Friendster method) (marius.tools.preprocess.datasets.livejournal.Livejournal method) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M method) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 method) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 method) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa method) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 method) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv method) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M method) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts method) (marius.tools.preprocess.datasets.twitter.Twitter method)

A

a_l (marius.nn.layers.GATLayer property) a_r (marius.nn.layers.GATLayer property) accumulateGradients() (marius.data.Batch method) activation (marius.config.LayerConfig property) (marius.tools.configuration.marius_config.LayerConfig attribute) ActivationFunction (class in marius.config) active_edges (marius.storage.GraphModelStorage property) active_in_memory_subgraph (marius.data.MariusGraph property) active_nodes (marius.storage.GraphModelStorage property) ADAGRAD (marius.config.OptimizerType attribute) AdagradOptimizer (class in marius.nn) AdagradOptions (class in marius.config) (class in marius.tools.configuration.datatypes) ADAM (marius.config.OptimizerType attribute) AdamOptimizer (class in marius.nn) AdamOptions (class in marius.config) (class in marius.tools.configuration.datatypes) add_metric() (marius.report.Reporter method) add_result() (marius.report.LinkPredictionReporter method) (marius.report.NodeClassificationReporter method) (marius.report.ProgressReporter method)

addFilterEdges() (marius.storage.GraphModelStorage method) aggregator (marius.config.GraphSageLayerOptions property) (marius.tools.configuration.datatypes.GraphSageLayerOptions attribute) ALL (marius.config.NeighborSamplingLayer attribute) ALL_BETA (marius.config.EdgeBucketOrdering attribute) all_read (marius.data.DataLoader property) all_reduce() (marius.nn.Model method) amsgrad (marius.config.AdamOptions property) (marius.nn.AdamOptimizer property) (marius.tools.configuration.datatypes.AdamOptions attribute) append() (marius.storage.FlatFile method) apply_relation() (marius.nn.decoders.edge.EdgeDecoder method) attention_dropout (marius.config.GATLayerOptions property) (marius.nn.layers.GATLayer property) (marius.tools.configuration.datatypes.GATLayerOptions attribute) average_heads (marius.config.GATLayerOptions property) (marius.tools.configuration.datatypes.GATLayerOptions attribute)

B

Batch (class in marius.data) batch_device_queue_size (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) batch_host_queue_size (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) batch_id (marius.data.Batch property) batch_id_offset (marius.data.DataLoader property) batch_loader_threads (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) batch_size (marius.config.EvaluationConfig property) (marius.config.TrainingConfig property) (marius.data.Batch property) (marius.tools.configuration.marius_config.EvaluationConfig attribute) (marius.tools.configuration.marius_config.TrainingConfig attribute) batch_transfer_threads (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) batches (marius.data.DataLoader property) batches_left (marius.data.DataLoader property) batches_processed (marius.data.DataLoader property) BCE_AFTER_SIGMOID (marius.config.LossFunctionType attribute) BCE_WITH_LOGITS (marius.config.LossFunctionType attribute) BCEAfterSigmoidLoss (class in marius.nn) BCEWithLogitsLoss (class in marius.nn)

beta_1 (marius.config.AdamOptions property) (marius.nn.AdamOptimizer property) (marius.tools.configuration.datatypes.AdamOptions attribute) beta_2 (marius.config.AdamOptions property) (marius.nn.AdamOptimizer property) (marius.tools.configuration.datatypes.AdamOptions attribute) bias (marius.config.LayerConfig property) (marius.nn.layers.Layer property) (marius.tools.configuration.marius_config.LayerConfig attribute) bias_init (marius.config.LayerConfig property) (marius.tools.configuration.marius_config.LayerConfig attribute) broadcast() (marius.nn.Model method) buffer_capacity (marius.config.PartitionBufferOptions property) (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) built-in function marius.nn.apply_activation() marius.nn.constant_init() marius.nn.glorot_normal() marius.nn.glorot_uniform() marius.nn.initialize_subtensor() marius.nn.initialize_tensor() marius.nn.normal_init() marius.nn.uniform_init() marius.storage.tensor_from_file()

C

checkpoint (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.TrainingConfig attribute) checkpoint_dir (marius.tools.configuration.marius_config.EvaluationConfig attribute) CheckpointConfig (class in marius.config) (class in marius.tools.configuration.marius_config) clear() (marius.data.Batch method) (marius.data.DENSEGraph method) (marius.data.MariusGraph method) (marius.report.LinkPredictionReporter method) (marius.report.NodeClassificationReporter method) (marius.report.ProgressReporter method) clear_grad() (marius.nn.Model method) (marius.nn.Optimizer method) clear_grad_all() (marius.nn.Model method) clearBatches() (marius.data.DataLoader method) Comparator (class in marius.nn.decoders.edge) comparator (marius.nn.decoders.edge.EdgeDecoder property) ComplEx (class in marius.nn.decoders.edge) COMPLEX (marius.config.DecoderType attribute) ComplexHadamardOperator (class in marius.nn.decoders.edge) compute_fans() (in module marius.nn) compute_labels() (marius.nn.decoders.node.NoOpNodeDecoder method)

compute_ranks() (marius.report.LinkPredictionReporter method) compute_scores() (marius.nn.decoders.edge.EdgeDecoder method) compute_threads (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) CONCAT (marius.config.ReductionLayerType attribute) ConcatReduction (class in marius.nn.layers) config (marius.nn.layers.Layer property) constant (marius.config.ConstantInitOptions property) CONSTANT (marius.config.InitDistribution attribute) constant (marius.tools.configuration.datatypes.ConstantInitOptions attribute) ConstantInitOptions (class in marius.config) (class in marius.tools.configuration.datatypes) CONV (marius.config.DenseLayerType attribute) copy() (marius.storage.FlatFile method) CORRUPT_NODE (marius.config.EdgeDecoderMethod attribute) CorruptNodeNegativeSampler (class in marius.data.samplers) CosineCompare (class in marius.nn.decoders.edge) CrossEntropyLoss (class in marius.nn) current_edge (marius.data.DataLoader property) current_subgraph_state (marius.storage.GraphModelStorage property) CUSTOM (marius.config.EdgeBucketOrdering attribute) (marius.config.NodePartitionOrdering attribute)

D

data (marius.storage.Storage property) DataLoader (class in marius.data) dataloader (marius.pipeline.Evaluator property) (marius.pipeline.GraphEncoder property) (marius.pipeline.Trainer property) Dataset (class in marius.tools.preprocess.dataset) dataset (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) dataset_dir (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) dataset_name (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute) dataset_url (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute) DatasetConfig (class in marius.config) (class in marius.tools.configuration.marius_config) Decoder (class in marius.nn.decoders) decoder (marius.config.ModelConfig property) (marius.nn.Model property) (marius.tools.configuration.marius_config.ModelConfig attribute) DecoderConfig (class in marius.config) (class in marius.tools.configuration.marius_config) DecoderOptions (class in marius.config) (class in marius.tools.configuration.datatypes) DecoderType (class in marius.config) DEFAULT (marius.config.OptimizerType attribute)

degree_fraction (marius.config.NegativeSamplingConfig property) (marius.data.samplers.CorruptNodeNegativeSampler property) (marius.tools.configuration.marius_config.NegativeSamplingConfig attribute) DENSE (marius.config.LayerType attribute) dense_graph (marius.data.Batch property) dense_optimizer (marius.config.ModelConfig property) (marius.tools.configuration.marius_config.ModelConfig attribute) DENSEGraph (class in marius.data) DenseLayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) DenseLayerType (class in marius.config) device (marius.nn.encoders.GeneralEncoder property) (marius.nn.layers.Layer property) (marius.nn.Model property) (marius.storage.Storage property) device_id (marius.data.Batch property) device_ids (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) DEVICE_MEMORY (marius.config.StorageBackend attribute) device_models (marius.nn.Model property) device_type (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) dim0_size (marius.storage.Storage property) dim1_size (marius.storage.Storage property) DISPERSED (marius.config.NodePartitionOrdering attribute) DistMult (class in marius.nn.decoders.edge) DISTMULT (marius.config.DecoderType attribute) DotCompare (class in marius.nn.decoders.edge) download() (marius.tools.preprocess.dataset.Dataset method) (marius.tools.preprocess.datasets.fb15k.FB15K method) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 method) (marius.tools.preprocess.datasets.freebase86m.Freebase86m method) (marius.tools.preprocess.datasets.friendster.Friendster method) (marius.tools.preprocess.datasets.livejournal.Livejournal method) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M method) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 method) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 method) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa method) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 method) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv method) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M method) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts method) (marius.tools.preprocess.datasets.twitter.Twitter method) DROPOUT (marius.config.NeighborSamplingLayer attribute) DropoutSamplingOptions (class in marius.config) (class in marius.tools.configuration.datatypes) dst_neg_filter (marius.data.Batch property) dst_neg_indices (marius.data.Batch property) dst_neg_indices_mapping (marius.data.Batch property) dst_sorted_edges (marius.data.MariusGraph property) dtype (marius.config.StorageOptions property) (marius.storage.Storage property) (marius.tools.configuration.datatypes.StorageOptions attribute)

E

edge_bucket_ordering (marius.config.PartitionBufferOptions property) (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) edge_bucket_sizes (marius.storage.Storage property) edge_buckets_per_buffer (marius.data.DataLoader property) edge_decoder_method (marius.tools.configuration.datatypes.EdgeDecoderOptions attribute) edge_features_file (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute) edge_file_name (marius.tools.configuration.constants.PathConstants attribute) edge_list_file (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute) edge_sampler (marius.data.DataLoader property) EdgeBucketOrdering (class in marius.config) EdgeDecoder (class in marius.nn.decoders.edge) EdgeDecoderMethod (class in marius.config) EdgeDecoderOptions (class in marius.config) (class in marius.tools.configuration.datatypes) edges (marius.config.StorageConfig property) (marius.data.Batch property) (marius.tools.configuration.marius_config.StorageConfig attribute) edges_directory (marius.tools.configuration.constants.PathConstants attribute)

edgeSample() (marius.data.DataLoader method) EMBEDDING (marius.config.LayerType attribute) embedding_dim (marius.tools.configuration.marius_config.EncoderConfig attribute) embedding_size (marius.nn.decoders.edge.EdgeDecoder property) EmbeddingLayer (class in marius.nn.layers) EmbeddingLayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) embeddings (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) embeddingsOffDevice() (marius.storage.GraphModelStorage method) embeddingsToHost() (marius.data.Batch method) ENCODE (marius.config.LearningTask attribute) encode() (marius.pipeline.GraphEncoder method) encoded_uniques (marius.data.Batch property) encoder (marius.config.ModelConfig property) (marius.nn.Model property) (marius.tools.configuration.marius_config.ModelConfig attribute) encoder_config (marius.nn.encoders.GeneralEncoder property) EncoderConfig (class in marius.config) (class in marius.tools.configuration.marius_config) epochComplete() (marius.data.DataLoader method) epochs_per_eval (marius.config.EvaluationConfig property) (marius.tools.configuration.marius_config.EvaluationConfig attribute) epochs_per_shuffle (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.TrainingConfig attribute) epochs_processed (marius.data.DataLoader property) eps (marius.config.AdagradOptions property) (marius.config.AdamOptions property) (marius.nn.AdagradOptimizer property) (marius.nn.AdamOptimizer property) (marius.tools.configuration.datatypes.AdagradOptions attribute) (marius.tools.configuration.datatypes.AdamOptions attribute) eval_neighbor_sampling (marius.config.EncoderConfig property) (marius.tools.configuration.marius_config.EncoderConfig attribute) evaluate() (marius.pipeline.Evaluator method) evaluate_batch() (marius.nn.Model method) evaluation (marius.config.MariusConfig property) (marius.tools.configuration.marius_config.MariusConfig attribute) evaluation_config (marius.data.DataLoader property) evaluation_negative_sampler (marius.data.DataLoader property) evaluation_neighbor_sampler (marius.data.DataLoader property) EvaluationConfig (class in marius.config) (class in marius.tools.configuration.marius_config) Evaluator (class in marius.pipeline) export_encoded_nodes (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute)

F

FB15K (class in marius.tools.preprocess.datasets.fb15k) FB15K237 (class in marius.tools.preprocess.datasets.fb15k_237) FEATURE (marius.config.LayerType attribute) FeatureLayer (class in marius.nn.layers) FeatureLayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) features (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) features_file_name (marius.tools.configuration.constants.PathConstants attribute) file_ext (marius.tools.configuration.constants.PathConstants attribute) filename (marius.storage.PartitionBufferStorage property) (marius.storage.Storage property) filtered (marius.config.NegativeSamplingConfig property) (marius.data.samplers.CorruptNodeNegativeSampler property) (marius.tools.configuration.marius_config.NegativeSamplingConfig attribute) fine_to_coarse_ratio (marius.config.PartitionBufferOptions property) (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) finishedBatch() (marius.data.DataLoader method) FLAT_FILE (marius.config.StorageBackend attribute)

FlatFile (class in marius.storage) forward() (marius.nn.decoders.node.NodeDecoder method) (marius.nn.encoders.GeneralEncoder method) (marius.nn.layers.ConcatReduction method) (marius.nn.layers.EmbeddingLayer method) (marius.nn.layers.FeatureLayer method) (marius.nn.layers.GATLayer method) (marius.nn.layers.GCNLayer method) (marius.nn.layers.GNNLayer method) (marius.nn.layers.GraphSageLayer method) (marius.nn.layers.LinearReduction method) (marius.nn.layers.ReductionLayer method) forward_lp() (marius.nn.Model method) forward_nc() (marius.nn.Model method) Freebase86m (class in marius.tools.preprocess.datasets.freebase86m) Friendster (class in marius.tools.preprocess.datasets.friendster) full_graph_evaluation (marius.config.EvaluationConfig property) (marius.config.StorageConfig property) (marius.storage.GraphModelStorage property) (marius.tools.configuration.marius_config.StorageConfig attribute)

G

GAT (marius.config.GNNLayerType attribute) GATLayer (class in marius.nn.layers) GATLayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) GCN (marius.config.GNNLayerType attribute) (marius.config.GraphSageAggregator attribute) GCNLayer (class in marius.nn.layers) GeneralEncoder (class in marius.nn.encoders) get_edges() (marius.storage.GraphModelStorage method) get_edges_range() (marius.storage.GraphModelStorage method) get_model_dir_path() (in module marius.tools.configuration.marius_config) getActivationFunction() (in module marius.config) getBatch() (marius.data.DataLoader method) getBatchesProcessed() (marius.data.DataLoader method) getDecoderType() (in module marius.config) getDenseLayerType() (in module marius.config) getDtype() (in module marius.config) getEdgeBucketOrderingEnum() (in module marius.config) getEdgeDecoderMethod() (in module marius.config) getEdges() (marius.data.MariusGraph method) getEpochsProcessed() (marius.data.DataLoader method) getGlobalToLocalMap() (marius.storage.PartitionBufferStorage method) getGNNLayerType() (in module marius.config) getGraphSageAggregator() (in module marius.config) getInitDistribution() (in module marius.config) getLayerOffset() (marius.data.DENSEGraph method) getLayerType() (in module marius.config) getLearningTask() (in module marius.config) getLossFunctionType() (in module marius.config) getLossReduction() (in module marius.config) getNeighborIDs() (marius.data.DENSEGraph method) getNeighborOffsets() (marius.data.MariusGraph method) getNeighborSamplingLayer() (in module marius.config) getNeighborsForNodeIds() (marius.data.MariusGraph method) getNextAdmit() (marius.storage.PartitionBufferStorage method) getNextBatch() (marius.data.DataLoader method) getNextEvict() (marius.storage.PartitionBufferStorage method) getNodeEmbeddings() (marius.storage.GraphModelStorage method) getNodeEmbeddingsRange() (marius.storage.GraphModelStorage method) getNodeEmbeddingState() (marius.storage.GraphModelStorage method) getNodeEmbeddingStateRange() (marius.storage.GraphModelStorage method) getNodeFeatures() (marius.storage.GraphModelStorage method) getNodeFeaturesRange() (marius.storage.GraphModelStorage method)

getNodeIdsRange() (marius.storage.GraphModelStorage method) getNodeLabels() (marius.storage.GraphModelStorage method) getNodeLabelsRange() (marius.storage.GraphModelStorage method) getNodePartitionOrderingEnum() (in module marius.config) getNumActiveEdges() (marius.storage.GraphModelStorage method) getNumActiveNodes() (marius.storage.GraphModelStorage method) getNumEdges() (marius.data.DataLoader method) (marius.storage.GraphModelStorage method) getNumInMemory() (marius.storage.PartitionBufferStorage method) getNumNeighbors() (marius.data.MariusGraph method) getNumNodes() (marius.storage.GraphModelStorage method) getNumNodesInMemory() (marius.storage.GraphModelStorage method) getNumPartitions() (marius.storage.GraphModelStorage method) getOptimizerType() (in module marius.config) getRandomNodeIds() (marius.storage.GraphModelStorage method) getReductionLayerType() (in module marius.config) getRelationIDs() (marius.data.MariusGraph method) getStorageBackend() (in module marius.config) GLOROT_NORMAL (marius.config.InitDistribution attribute) GLOROT_UNIFORM (marius.config.InitDistribution attribute) GNN (marius.config.LayerType attribute) GNNLayer (class in marius.nn.layers) GNNLayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) GNNLayerType (class in marius.config) gpu_model_average (marius.tools.configuration.marius_config.PipelineConfig attribute) gpu_sync_interval (marius.tools.configuration.marius_config.PipelineConfig attribute) gradient_transfer_threads (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) gradient_update_threads (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) gradients_device_queue_size (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) gradients_host_queue_size (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) GRAPH_SAGE (marius.config.GNNLayerType attribute) graph_storage (marius.data.DataLoader property) GraphEncoder (class in marius.pipeline) GraphModelStorage (class in marius.storage) GraphSageAggregator (class in marius.config) GraphSageLayer (class in marius.nn.layers) GraphSageLayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes)

H

HadamardOperator (class in marius.nn.decoders.edge) hasNextBatch() (marius.data.DataLoader method) hasSwap() (marius.storage.GraphModelStorage method) (marius.storage.PartitionBufferStorage method)

head_dim (marius.nn.layers.GATLayer property) hop_offsets (marius.data.DENSEGraph property) HOST_MEMORY (marius.config.StorageBackend attribute)

I

in_neighbors (marius.data.DENSEGraph property) in_neighbors_vec (marius.data.DENSEGraph property) in_num_neighbors (marius.data.MariusGraph property) in_offsets (marius.data.MariusGraph property) in_sorted_uniques (marius.data.MariusGraph property) indexAdd() (marius.storage.Storage method) indexPut() (marius.storage.Storage method) indexRead() (marius.storage.Storage method) infer_model_dir() (in module marius.tools.configuration.marius_config) init (marius.config.LayerConfig property) (marius.tools.configuration.marius_config.LayerConfig attribute) init_bias() (marius.nn.layers.Layer method) init_embeddings() (marius.nn.layers.EmbeddingLayer method) init_subgraph() (marius.storage.GraphModelStorage method) init_value (marius.config.AdagradOptions property) (marius.nn.AdagradOptimizer property) (marius.tools.configuration.datatypes.AdagradOptions attribute) InitConfig (class in marius.config) (class in marius.tools.configuration.marius_config) InitDistribution (class in marius.config)

initialize_model_dir() (in module marius.tools.configuration.marius_config) initializeBatches() (marius.data.DataLoader method) initialized (marius.storage.Storage property) (marius.tools.configuration.marius_config.DatasetConfig attribute) InitOptions (class in marius.config) (class in marius.tools.configuration.datatypes) InMemory (class in marius.storage) input_dim (marius.config.EdgeDecoderOptions property) (marius.config.LayerConfig property) (marius.nn.layers.GNNLayer property) (marius.tools.configuration.marius_config.LayerConfig attribute) input_dropout (marius.config.GATLayerOptions property) (marius.nn.layers.GATLayer property) (marius.tools.configuration.datatypes.GATLayerOptions attribute) interval (marius.config.CheckpointConfig property) (marius.tools.configuration.marius_config.CheckpointConfig attribute) inverse_edges (marius.config.EdgeDecoderOptions property) (marius.tools.configuration.datatypes.EdgeDecoderOptions attribute) inverse_relations (marius.nn.decoders.edge.EdgeDecoder property) isTrain() (marius.data.DataLoader method)

L

L2Compare (class in marius.nn.decoders.edge) labels_file_name (marius.tools.configuration.constants.PathConstants attribute) labels_path (marius.tools.configuration.constants.PathConstants attribute) Layer (class in marius.nn.layers) LayerConfig (class in marius.config) (class in marius.tools.configuration.marius_config) LayeredNeighborSampler (class in marius.data.samplers) LayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) layers (marius.config.EncoderConfig property) (marius.nn.encoders.GeneralEncoder property) (marius.tools.configuration.marius_config.EncoderConfig attribute) LayerType (class in marius.config) learning_rate (marius.config.OptimizerOptions property) (marius.nn.AdagradOptimizer property) (marius.nn.AdamOptimizer property) (marius.nn.SGDOptimizer property) (marius.tools.configuration.datatypes.AdagradOptions attribute) (marius.tools.configuration.datatypes.AdamOptions attribute) (marius.tools.configuration.datatypes.OptimizerOptions attribute) learning_task (marius.config.ModelConfig property) (marius.nn.decoders.Decoder property) (marius.nn.Model property) (marius.pipeline.Trainer property) (marius.tools.configuration.marius_config.ModelConfig attribute) LearningTask (class in marius.config) LINEAR (marius.config.DenseLayerType attribute) (marius.config.ReductionLayerType attribute) LinearReduction (class in marius.nn.layers) LINK_PREDICTION (marius.config.LearningTask attribute)

LinkPredictionReporter (class in marius.report) Livejournal (class in marius.tools.preprocess.datasets.livejournal) load() (marius.nn.Model method) (marius.storage.GraphModelStorage method) (marius.storage.Storage method) load_config() (in module marius.tools.configuration.marius_config) loadConfig() (in module marius.config) loadCPUParameters() (marius.data.DataLoader method) loaded (marius.storage.PartitionBufferStorage property) loadGPUParameters() (marius.data.DataLoader method) loadStorage() (marius.data.DataLoader method) local_filter_mode (marius.tools.configuration.marius_config.NegativeSamplingConfig attribute) log_level (marius.tools.configuration.marius_config.StorageConfig attribute) logs_per_epoch (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.TrainingConfig attribute) loss (marius.config.ModelConfig property) (marius.tools.configuration.marius_config.ModelConfig attribute) loss_function (marius.nn.Model property) loss_reduction (marius.config.LossOptions property) (marius.config.RankingLossOptions property) LossConfig (class in marius.config) (class in marius.tools.configuration.marius_config) LossFunction (class in marius.nn) LossFunctionType (class in marius.config) LossOptions (class in marius.config) (class in marius.tools.configuration.datatypes) LossReduction (class in marius.config) lr_decay (marius.config.AdagradOptions property) (marius.nn.AdagradOptimizer property) (marius.tools.configuration.datatypes.AdagradOptions attribute)

M

margin (marius.config.RankingLossOptions property) (marius.tools.configuration.datatypes.RankingLossOptions attribute) marius.config module marius.nn.apply_activation() built-in function marius.nn.constant_init() built-in function marius.nn.glorot_normal() built-in function marius.nn.glorot_uniform() built-in function marius.nn.initialize_subtensor() built-in function marius.nn.initialize_tensor() built-in function marius.nn.normal_init() built-in function marius.nn.uniform_init() built-in function marius.storage.tensor_from_file() built-in function marius.tools.configuration.constants module marius.tools.configuration.datatypes module marius.tools.configuration.marius_config module marius.tools.preprocess.converters module marius.tools.preprocess.converters.partitioners module marius.tools.preprocess.converters.readers module marius.tools.preprocess.converters.writers module MariusConfig (class in marius.config) (class in marius.tools.configuration.marius_config) MariusGraph (class in marius.data) max_in_num_neighbors_ (marius.data.MariusGraph property) max_neighbors (marius.config.UniformSamplingOptions property) (marius.tools.configuration.datatypes.UniformSamplingOptions attribute) max_out_num_neighbors_ (marius.data.MariusGraph property) MEAN (marius.config.GraphSageAggregator attribute) (marius.config.LossReduction attribute)

mean (marius.config.NormalInitOptions property) (marius.tools.configuration.datatypes.NormalInitOptions attribute) mem_load() (marius.storage.FlatFile method) mem_unload() (marius.storage.FlatFile method) merge() (marius.tools.configuration.marius_config.CheckpointConfig method) (marius.tools.configuration.marius_config.DatasetConfig method) (marius.tools.configuration.marius_config.DecoderConfig method) (marius.tools.configuration.marius_config.EncoderConfig method) (marius.tools.configuration.marius_config.EvaluationConfig method) (marius.tools.configuration.marius_config.InitConfig method) (marius.tools.configuration.marius_config.LayerConfig method) (marius.tools.configuration.marius_config.LossConfig method) (marius.tools.configuration.marius_config.ModelConfig method) (marius.tools.configuration.marius_config.NegativeSamplingConfig method) (marius.tools.configuration.marius_config.NeighborSamplingConfig method) (marius.tools.configuration.marius_config.OptimizerConfig method) (marius.tools.configuration.marius_config.PipelineConfig method) (marius.tools.configuration.marius_config.StorageBackendConfig method) (marius.tools.configuration.marius_config.StorageConfig method) (marius.tools.configuration.marius_config.TrainingConfig method) metrics (marius.report.Reporter property) mode (marius.config.EdgeDecoderOptions property) (marius.nn.decoders.edge.EdgeDecoder property) Model (class in marius.nn) model (marius.config.MariusConfig property) (marius.tools.configuration.marius_config.MariusConfig attribute) model_dir (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) model_file (marius.tools.configuration.constants.PathConstants attribute) model_state_file (marius.tools.configuration.constants.PathConstants attribute) ModelConfig (class in marius.config) (class in marius.tools.configuration.marius_config) module marius.config marius.tools.configuration.constants marius.tools.configuration.datatypes marius.tools.configuration.marius_config marius.tools.preprocess.converters marius.tools.preprocess.converters.partitioners marius.tools.preprocess.converters.readers marius.tools.preprocess.converters.writers move() (marius.storage.FlatFile method) MSE (marius.config.LossFunctionType attribute) MSELoss (class in marius.nn)

N

name (marius.config.ActivationFunction property) (marius.config.DecoderType property) (marius.config.DenseLayerType property) (marius.config.EdgeBucketOrdering property) (marius.config.EdgeDecoderMethod property) (marius.config.GNNLayerType property) (marius.config.GraphSageAggregator property) (marius.config.InitDistribution property) (marius.config.LayerType property) (marius.config.LearningTask property) (marius.config.LossFunctionType property) (marius.config.LossReduction property) (marius.config.NeighborSamplingLayer property) (marius.config.NodePartitionOrdering property) (marius.config.OptimizerType property) (marius.config.ReductionLayerType property) (marius.config.StorageBackend property) neg_edges (marius.data.Batch property) negative_sampler (marius.data.DataLoader property) negative_sampling (marius.config.EvaluationConfig property) (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.EvaluationConfig attribute) (marius.tools.configuration.marius_config.TrainingConfig attribute) negative_slope (marius.config.GATLayerOptions property) (marius.tools.configuration.datatypes.GATLayerOptions attribute) negatives_per_positive (marius.config.NegativeSamplingConfig property) (marius.tools.configuration.marius_config.NegativeSamplingConfig attribute) NegativeSamplingConfig (class in marius.config) (class in marius.tools.configuration.marius_config) neighbor_sampler (marius.data.DataLoader property) NeighborSamplingConfig (class in marius.config) (class in marius.tools.configuration.marius_config) NeighborSamplingLayer (class in marius.config) NeighborSamplingOptions (class in marius.config) (class in marius.tools.configuration.datatypes) NEW_BETA (marius.config.EdgeBucketOrdering attribute) next_subgraph_state (marius.storage.GraphModelStorage property) nextEpoch() (marius.data.DataLoader method) NODE (marius.config.DecoderType attribute) NODE_CLASSIFICATION (marius.config.LearningTask attribute) node_embeddings (marius.data.Batch property) node_embeddings_file_name (marius.tools.configuration.constants.PathConstants attribute) node_embeddings_state (marius.data.Batch property) node_embeddings_state_file_name (marius.tools.configuration.constants.PathConstants attribute) node_feature_dim (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) node_features (marius.data.Batch property) node_features_file (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute) node_features_path (marius.tools.configuration.constants.PathConstants attribute) node_file_name (marius.tools.configuration.constants.PathConstants attribute) node_gradients (marius.data.Batch property) node_ids (marius.data.MariusGraph property) node_ids_per_buffer (marius.data.DataLoader property) node_labels (marius.data.Batch property) node_mapping_file (marius.tools.configuration.constants.PathConstants attribute) (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute)

node_mapping_path (marius.tools.configuration.constants.PathConstants attribute) node_partition_ordering (marius.config.PartitionBufferOptions property) (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) node_properties (marius.data.DENSEGraph property) node_state_update (marius.data.Batch property) node_type_file (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute) NodeClassificationReporter (class in marius.report) NodeDecoder (class in marius.nn.decoders.node) NodePartitionOrdering (class in marius.config) nodes (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) nodes_directory (marius.tools.configuration.constants.PathConstants attribute) nodeSample() (marius.data.DataLoader method) NONE (marius.config.ActivationFunction attribute) (marius.config.DenseLayerType attribute) (marius.config.GNNLayerType attribute) (marius.config.LayerType attribute) (marius.config.ReductionLayerType attribute) NoOp (class in marius.nn.decoders.edge) NoOpNodeDecoder (class in marius.nn.decoders.node) NORMAL (marius.config.InitDistribution attribute) NormalInitOptions (class in marius.config) (class in marius.tools.configuration.datatypes) num_cache_partitions (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) num_chunks (marius.config.NegativeSamplingConfig property) (marius.data.samplers.CorruptNodeNegativeSampler property) (marius.tools.configuration.marius_config.NegativeSamplingConfig attribute) num_classes (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) num_edges (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) num_epochs (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.TrainingConfig attribute) num_heads (marius.config.GATLayerOptions property) (marius.tools.configuration.datatypes.GATLayerOptions attribute) num_negatives (marius.data.samplers.CorruptNodeNegativeSampler property) num_nodes (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) num_nodes_in_memory (marius.data.DENSEGraph property) (marius.data.MariusGraph property) num_partitions (marius.config.PartitionBufferOptions property) (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) num_relations (marius.config.DatasetConfig property) (marius.nn.decoders.edge.EdgeDecoder property) (marius.nn.encoders.GeneralEncoder property) (marius.tools.configuration.marius_config.DatasetConfig attribute) num_steps (marius.nn.Optimizer property) num_test (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) num_train (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) num_valid (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute)

O

offset (marius.nn.layers.EmbeddingLayer property) (marius.nn.layers.FeatureLayer property) OGBLCitation2 (class in marius.tools.preprocess.datasets.ogbl_citation2) OGBLPpa (class in marius.tools.preprocess.datasets.ogbl_ppa) OGBLWikiKG2 (class in marius.tools.preprocess.datasets.ogbl_wikikg2) OGBMag240M (class in marius.tools.preprocess.datasets.ogb_mag240m) OGBNArxiv (class in marius.tools.preprocess.datasets.ogbn_arxiv) OGBNPapers100M (class in marius.tools.preprocess.datasets.ogbn_papers100m) OGBNProducts (class in marius.tools.preprocess.datasets.ogbn_products) OGBWikiKG90Mv2 (class in marius.tools.preprocess.datasets.ogb_wikikg90mv2) OLD_BETA (marius.config.EdgeBucketOrdering attribute) ONES (marius.config.InitDistribution attribute) ONLY_POS (marius.config.EdgeDecoderMethod attribute) Optimizer (class in marius.nn) optimizer (marius.config.DecoderConfig property) (marius.config.LayerConfig property) (marius.tools.configuration.marius_config.DecoderConfig attribute) (marius.tools.configuration.marius_config.LayerConfig attribute) OptimizerConfig (class in marius.config) (class in marius.tools.configuration.marius_config) OptimizerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) optimizers (marius.nn.Model property) OptimizerType (class in marius.config) options (marius.config.DecoderConfig property) (marius.config.InitConfig property) (marius.config.LayerConfig property) (marius.config.LossConfig property) (marius.config.NeighborSamplingConfig property) (marius.config.OptimizerConfig property) (marius.config.StorageBackendConfig property) (marius.nn.layers.GATLayer property) (marius.nn.layers.GCNLayer property) (marius.nn.layers.GraphSageLayer property) (marius.storage.PartitionBufferStorage property) (marius.tools.configuration.marius_config.DecoderConfig attribute) (marius.tools.configuration.marius_config.InitConfig attribute) (marius.tools.configuration.marius_config.LayerConfig attribute) (marius.tools.configuration.marius_config.LossConfig attribute) (marius.tools.configuration.marius_config.NeighborSamplingConfig attribute) (marius.tools.configuration.marius_config.OptimizerConfig attribute) (marius.tools.configuration.marius_config.StorageBackendConfig attribute)

out_neighbors (marius.data.DENSEGraph property) out_neighbors_vec (marius.data.DENSEGraph property) out_num_neighbors (marius.data.MariusGraph property) out_offsets (marius.data.MariusGraph property) out_sorted_uniques (marius.data.MariusGraph property) output_dim (marius.config.LayerConfig property) (marius.nn.layers.GNNLayer property) (marius.tools.configuration.marius_config.LayerConfig attribute) output_directory (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute)

P

PARTITION_BUFFER (marius.config.StorageBackend attribute) partition_offsets_file (marius.tools.configuration.constants.PathConstants attribute) PartitionBufferOptions (class in marius.config) (class in marius.tools.configuration.datatypes) PartitionBufferStorage (class in marius.storage) PathConstants (class in marius.tools.configuration.constants) performMap() (marius.data.DENSEGraph method) performNextSwap() (marius.storage.PartitionBufferStorage method) performSwap() (marius.storage.GraphModelStorage method) pipeline (marius.config.EvaluationConfig property) (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.EvaluationConfig attribute) (marius.tools.configuration.marius_config.TrainingConfig attribute) PipelineConfig (class in marius.config) (class in marius.tools.configuration.marius_config) PipelineEncoder (class in marius.pipeline) PipelineEvaluator (class in marius.pipeline) PipelineTrainer (class in marius.pipeline) populate_dataset_stats() (marius.tools.configuration.marius_config.DatasetConfig method) post_hook() (marius.nn.layers.Layer method) prefetch (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute)

prefetching (marius.config.PartitionBufferOptions property) (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) prepareForNextLayer() (marius.data.DENSEGraph method) preprocess() (marius.tools.preprocess.dataset.Dataset method) (marius.tools.preprocess.datasets.fb15k.FB15K method) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 method) (marius.tools.preprocess.datasets.freebase86m.Freebase86m method) (marius.tools.preprocess.datasets.friendster.Friendster method) (marius.tools.preprocess.datasets.livejournal.Livejournal method) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M method) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 method) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 method) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa method) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 method) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv method) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M method) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts method) (marius.tools.preprocess.datasets.twitter.Twitter method) progress_reporter (marius.pipeline.GraphEncoder property) (marius.pipeline.Trainer property) ProgressReporter (class in marius.report)

R

random_seed (marius.config.ModelConfig property) (marius.tools.configuration.marius_config.ModelConfig attribute) RandomEdgeSampler (class in marius.data.samplers) randomly_assign_edge_buckets (marius.tools.configuration.datatypes.PartitionBufferOptions attribute) range() (marius.storage.Storage method) rangePut() (marius.storage.Storage method) RANKING (marius.config.LossFunctionType attribute) RankingLoss (class in marius.nn) RankingLossOptions (class in marius.config) (class in marius.tools.configuration.datatypes) rate (marius.config.DropoutSamplingOptions property) (marius.tools.configuration.datatypes.DropoutSamplingOptions attribute) read_edge_bucket_sizes() (marius.storage.Storage method) REDUCTION (marius.config.LayerType attribute) reduction (marius.tools.configuration.datatypes.LossOptions attribute) ReductionLayer (class in marius.nn.layers) ReductionLayerOptions (class in marius.config) (class in marius.tools.configuration.datatypes) ReductionLayerType (class in marius.config) rel_feature_dim (marius.config.DatasetConfig property) (marius.tools.configuration.marius_config.DatasetConfig attribute) rel_neg_indices (marius.data.Batch property) relation_features_file (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute) relation_features_path (marius.tools.configuration.constants.PathConstants attribute) relation_mapping_file (marius.tools.configuration.constants.PathConstants attribute) (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute)

relation_mapping_path (marius.tools.configuration.constants.PathConstants attribute) relation_operator (marius.nn.decoders.edge.EdgeDecoder property) RelationOperator (class in marius.nn.decoders.edge) relations (marius.nn.decoders.edge.EdgeDecoder property) RELU (marius.config.ActivationFunction attribute) report() (marius.report.Reporter method) Reporter (class in marius.report) reporter (marius.nn.Model property) reset() (marius.nn.decoders.edge.ComplEx method) (marius.nn.decoders.edge.DistMult method) (marius.nn.decoders.edge.TransE method) (marius.nn.decoders.node.NoOpNodeDecoder method) (marius.nn.encoders.GeneralEncoder method) (marius.nn.layers.ConcatReduction method) (marius.nn.layers.EmbeddingLayer method) (marius.nn.layers.FeatureLayer method) (marius.nn.layers.GATLayer method) (marius.nn.layers.GCNLayer method) (marius.nn.layers.GraphSageLayer method) (marius.nn.layers.LinearReduction method) reset_state() (marius.nn.Optimizer method) resume_from_checkpoint (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.TrainingConfig attribute) resume_training (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.TrainingConfig attribute) RGCN (marius.config.GNNLayerType attribute) root_node_indices (marius.data.Batch property)

S

sampling_layers (marius.data.samplers.LayeredNeighborSampler property) save() (marius.nn.Model method) (marius.report.LinkPredictionReporter method) (marius.report.NodeClassificationReporter method) save_best (marius.config.CheckpointConfig property) (marius.tools.configuration.marius_config.CheckpointConfig attribute) save_model (marius.config.TrainingConfig property) (marius.tools.configuration.marius_config.TrainingConfig attribute) save_state (marius.config.CheckpointConfig property) (marius.tools.configuration.marius_config.CheckpointConfig attribute) saved_full_config_file_name (marius.tools.configuration.constants.PathConstants attribute) scale_factor (marius.config.UniformInitOptions property) (marius.tools.configuration.datatypes.UniformInitOptions attribute) select_relations() (marius.nn.decoders.edge.EdgeDecoder method) SEQUENTIAL (marius.config.NodePartitionOrdering attribute) set_edge_storage() (marius.storage.GraphModelStorage method) set_node_storage() (marius.storage.GraphModelStorage method) setActiveEdges() (marius.data.DataLoader method) (marius.storage.GraphModelStorage method) setActiveNodes() (marius.data.DataLoader method) (marius.storage.GraphModelStorage method) setBufferOrdering() (marius.data.DataLoader method) (marius.storage.GraphModelStorage method) (marius.storage.PartitionBufferStorage method) setFilterEdges() (marius.storage.GraphModelStorage method) setNodeProperties() (marius.data.DENSEGraph method) setTestSet() (marius.data.DataLoader method) (marius.storage.GraphModelStorage method) setTrainSet() (marius.data.DataLoader method) (marius.storage.GraphModelStorage method) setValidationSet() (marius.data.DataLoader method) (marius.storage.GraphModelStorage method) SGD (marius.config.OptimizerType attribute) SGDOptimizer (class in marius.nn) shuffle() (marius.storage.Storage method) shuffle_input (marius.config.StorageConfig property) (marius.tools.configuration.marius_config.StorageConfig attribute) shuffleEdges() (marius.storage.GraphModelStorage method) SIGMOID (marius.config.ActivationFunction attribute) SOFTMAX_CE (marius.config.LossFunctionType attribute) SoftmaxCrossEntropy (class in marius.nn) SOFTPLUS (marius.config.LossFunctionType attribute) SoftPlusLoss (class in marius.nn) sort() (marius.storage.Storage method) sort_all_edges() (marius.storage.GraphModelStorage method) spark (marius.tools.preprocess.dataset.Dataset attribute) (marius.tools.preprocess.datasets.fb15k.FB15K attribute) (marius.tools.preprocess.datasets.fb15k_237.FB15K237 attribute) (marius.tools.preprocess.datasets.freebase86m.Freebase86m attribute) (marius.tools.preprocess.datasets.friendster.Friendster attribute) (marius.tools.preprocess.datasets.livejournal.Livejournal attribute) (marius.tools.preprocess.datasets.ogb_mag240m.OGBMag240M attribute) (marius.tools.preprocess.datasets.ogb_wikikg90mv2.OGBWikiKG90Mv2 attribute) (marius.tools.preprocess.datasets.ogbl_citation2.OGBLCitation2 attribute) (marius.tools.preprocess.datasets.ogbl_ppa.OGBLPpa attribute) (marius.tools.preprocess.datasets.ogbl_wikikg2.OGBLWikiKG2 attribute) (marius.tools.preprocess.datasets.ogbn_arxiv.OGBNArxiv attribute) (marius.tools.preprocess.datasets.ogbn_papers100m.OGBNPapers100M attribute) (marius.tools.preprocess.datasets.ogbn_products.OGBNProducts attribute) (marius.tools.preprocess.datasets.twitter.Twitter attribute)

sparse_lr (marius.nn.Model property) sparse_optimizer (marius.config.ModelConfig property) (marius.tools.configuration.marius_config.ModelConfig attribute) src_neg_filter (marius.data.Batch property) src_neg_indices (marius.data.Batch property) src_neg_indices_mapping (marius.data.Batch property) src_sorted_edges (marius.data.MariusGraph property) staleness_bound (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) start_idx (marius.data.Batch property) status (marius.data.Batch property) std (marius.config.NormalInitOptions property) (marius.tools.configuration.datatypes.NormalInitOptions attribute) step() (marius.nn.Model method) (marius.nn.Optimizer method) step_all() (marius.nn.Model method) Storage (class in marius.storage) storage (marius.config.MariusConfig property) (marius.tools.configuration.marius_config.MariusConfig attribute) storage_ptrs (marius.storage.GraphModelStorage property) StorageBackend (class in marius.config) StorageBackendConfig (class in marius.config) (class in marius.tools.configuration.marius_config) StorageConfig (class in marius.config) (class in marius.tools.configuration.marius_config) StorageOptions (class in marius.config) (class in marius.tools.configuration.datatypes) SUM (marius.config.LossReduction attribute) SUPPORTED_EDGE_BACKENDS (marius.tools.configuration.marius_config.StorageConfig attribute) SUPPORTED_EMBEDDING_BACKENDS (marius.tools.configuration.marius_config.StorageConfig attribute) SUPPORTED_NODE_BACKENDS (marius.tools.configuration.marius_config.StorageConfig attribute) sync (marius.config.PipelineConfig property) (marius.tools.configuration.marius_config.PipelineConfig attribute) sync() (marius.storage.PartitionBufferStorage method) SynchronousEncoder (class in marius.pipeline) SynchronousEvaluator (class in marius.pipeline) SynchronousTrainer (class in marius.pipeline)

T

tensor_options (marius.nn.decoders.edge.EdgeDecoder property) test_edge_buckets_path (marius.tools.configuration.constants.PathConstants attribute) test_edges_path (marius.tools.configuration.constants.PathConstants attribute) test_file_prefix (marius.tools.configuration.constants.PathConstants attribute) test_nodes_path (marius.tools.configuration.constants.PathConstants attribute) to() (marius.data.Batch method) (marius.data.DENSEGraph method) (marius.data.MariusGraph method) train (marius.data.Batch property) (marius.data.DataLoader property) train() (marius.pipeline.Trainer method) train_batch() (marius.nn.Model method) train_edge_buckets_path (marius.tools.configuration.constants.PathConstants attribute) train_edges_path (marius.tools.configuration.constants.PathConstants attribute) train_neighbor_sampling (marius.config.EncoderConfig property) (marius.tools.configuration.marius_config.EncoderConfig attribute) train_nodes_path (marius.tools.configuration.constants.PathConstants attribute) Trainer (class in marius.pipeline) training (marius.config.MariusConfig property) (marius.tools.configuration.marius_config.MariusConfig attribute) training_config (marius.data.DataLoader property) training_file_prefix (marius.tools.configuration.constants.PathConstants attribute) training_negative_sampler (marius.data.DataLoader property) training_neighbor_sampler (marius.data.DataLoader property) TrainingConfig (class in marius.config) (class in marius.tools.configuration.marius_config) TransE (class in marius.nn.decoders.edge)

TRANSE (marius.config.DecoderType attribute) TranslationOperator (class in marius.nn.decoders.edge) Twitter (class in marius.tools.preprocess.datasets.twitter) TWO_LEVEL_BETA (marius.config.EdgeBucketOrdering attribute) type (marius.config.DecoderConfig property) (marius.config.DenseLayerOptions property) (marius.config.GNNLayerOptions property) (marius.config.InitConfig property) (marius.config.LayerConfig property) (marius.config.LossConfig property) (marius.config.NeighborSamplingConfig property) (marius.config.OptimizerConfig property) (marius.config.ReductionLayerOptions property) (marius.config.StorageBackendConfig property) (marius.tools.configuration.datatypes.DenseLayerOptions attribute) (marius.tools.configuration.datatypes.GATLayerOptions attribute) (marius.tools.configuration.datatypes.GNNLayerOptions attribute) (marius.tools.configuration.datatypes.GraphSageLayerOptions attribute) (marius.tools.configuration.datatypes.ReductionLayerOptions attribute) (marius.tools.configuration.marius_config.DecoderConfig attribute) (marius.tools.configuration.marius_config.InitConfig attribute) (marius.tools.configuration.marius_config.LayerConfig attribute) (marius.tools.configuration.marius_config.LossConfig attribute) (marius.tools.configuration.marius_config.NeighborSamplingConfig attribute) (marius.tools.configuration.marius_config.OptimizerConfig attribute) (marius.tools.configuration.marius_config.StorageBackendConfig attribute) type_safe_merge() (in module marius.tools.configuration.marius_config)

U

UNIFORM (marius.config.InitDistribution attribute) (marius.config.NeighborSamplingLayer attribute) UniformInitOptions (class in marius.config) (class in marius.tools.configuration.datatypes) UniformSamplingOptions (class in marius.config) (class in marius.tools.configuration.datatypes) unique_node_indices (marius.data.Batch property) unload() (marius.storage.GraphModelStorage method) (marius.storage.Storage method) unloadStorage() (marius.data.DataLoader method) update_subgraph() (marius.storage.GraphModelStorage method) updateAddNodeEmbeddings() (marius.storage.GraphModelStorage method)

updateAddNodeEmbeddingState() (marius.storage.GraphModelStorage method) updateEmbeddings() (marius.data.DataLoader method) updatePutNodeEmbeddings() (marius.storage.GraphModelStorage method) updatePutNodeEmbeddingState() (marius.storage.GraphModelStorage method) use_hashmap_sets (marius.config.NeighborSamplingConfig property) (marius.tools.configuration.marius_config.NeighborSamplingConfig attribute) use_incoming_nbrs (marius.config.NeighborSamplingConfig property) (marius.tools.configuration.marius_config.NeighborSamplingConfig attribute) use_inverse_relations (marius.nn.decoders.edge.EdgeDecoder property) use_outgoing_nbrs (marius.config.NeighborSamplingConfig property) (marius.tools.configuration.marius_config.NeighborSamplingConfig attribute) use_relation_features (marius.tools.configuration.datatypes.EdgeDecoderOptions attribute) useInMemorySubGraph() (marius.storage.GraphModelStorage method)

V

valid_edge_buckets_path (marius.tools.configuration.constants.PathConstants attribute) valid_edges_path (marius.tools.configuration.constants.PathConstants attribute) valid_nodes_path (marius.tools.configuration.constants.PathConstants attribute) validation_file_prefix (marius.tools.configuration.constants.PathConstants attribute) value (marius.config.ActivationFunction property) (marius.config.DecoderType property) (marius.config.DenseLayerType property) (marius.config.EdgeBucketOrdering property) (marius.config.EdgeDecoderMethod property) (marius.config.GNNLayerType property) (marius.config.GraphSageAggregator property) (marius.config.InitDistribution property) (marius.config.LayerType property) (marius.config.LearningTask property) (marius.config.LossFunctionType property) (marius.config.LossReduction property) (marius.config.NeighborSamplingLayer property) (marius.config.NodePartitionOrdering property) (marius.config.OptimizerType property) (marius.config.ReductionLayerType property) (marius.config.StorageBackend property)

W

w1 (marius.nn.layers.GraphSageLayer property) w2_ (marius.nn.layers.GraphSageLayer property) w_ (marius.nn.layers.GCNLayer property) weight_decay (marius.config.AdagradOptions property) (marius.config.AdamOptions property) (marius.nn.AdagradOptimizer property) (marius.nn.AdamOptimizer property) (marius.tools.configuration.datatypes.AdagradOptions attribute) (marius.tools.configuration.datatypes.AdamOptions attribute)

weight_matrices (marius.nn.layers.GATLayer property) weight_matrix (marius.nn.layers.LinearReduction property) without_replacement (marius.data.samplers.RandomEdgeSampler property) write() (marius.storage.Storage method)

Z

ZEROS (marius.config.InitDistribution attribute)