GraphStorage

class marius.storage.GraphModelStorage

__init__(self: marius._storage.GraphModelStorage, storage_ptrs: marius._storage.GraphModelStoragePtrs, storage_config: marius._config.StorageConfig) → None

__init__(self: marius._storage.GraphModelStorage, edges: marius._storage.Storage, nodes: marius._storage.Storage = None, node_features: marius._storage.Storage = None, node_embeddings: marius._storage.Storage = None, node_optim_state: marius._storage.Storage = None, node_labels: marius._storage.Storage = None, filter_edges: List[marius._storage.Storage] = [], train: bool = False, prefetch: bool = False) → None

getNodeEmbeddingState(self: marius._storage.GraphModelStorage, indices: torch.Tensor) → torch.Tensor

getNodeEmbeddingStateRange(self: marius._storage.GraphModelStorage, start: int, size: int) → torch.Tensor

getNodeEmbeddings(self: marius._storage.GraphModelStorage, indices: torch.Tensor) → torch.Tensor

getNodeEmbeddingsRange(self: marius._storage.GraphModelStorage, start: int, size: int) → torch.Tensor

getNodeFeatures(self: marius._storage.GraphModelStorage, indices: torch.Tensor) → torch.Tensor

getNodeFeaturesRange(self: marius._storage.GraphModelStorage, start: int, size: int) → torch.Tensor

getNodeIdsRange(self: marius._storage.GraphModelStorage, start: int, size: int) → torch.Tensor

getNodeLabels(self: marius._storage.GraphModelStorage, indices: torch.Tensor) → torch.Tensor

getNodeLabelsRange(self: marius._storage.GraphModelStorage, start: int, size: int) → torch.Tensor

getRandomNodeIds(self: marius._storage.GraphModelStorage, size: int) → torch.Tensor

get_edges(self: marius._storage.GraphModelStorage, indices: torch.Tensor) → torch.Tensor

get_edges_range(self: marius._storage.GraphModelStorage, start: int, size: int) → torch.Tensor

init_subgraph(self: marius._storage.GraphModelStorage, buffer_state: torch.Tensor) → None

setActiveEdges(self: marius._storage.GraphModelStorage, active_edges: torch.Tensor) → None

setActiveNodes(self: marius._storage.GraphModelStorage, node_ids: torch.Tensor) → None

setBufferOrdering(self: marius._storage.GraphModelStorage, buffer_states: List[torch.Tensor]) → None

updateAddNodeEmbeddingState(self: marius._storage.GraphModelStorage, indices: torch.Tensor, values: torch.Tensor) → None

updateAddNodeEmbeddings(self: marius._storage.GraphModelStorage, indices: torch.Tensor, values: torch.Tensor) → None

updatePutNodeEmbeddingState(self: marius._storage.GraphModelStorage, indices: torch.Tensor, state: torch.Tensor) → None

updatePutNodeEmbeddings(self: marius._storage.GraphModelStorage, indices: torch.Tensor, embeddings: torch.Tensor) → None

property active_edges

property active_nodes

addFilterEdges(self: marius._storage.GraphModelStorage, arg0: marius._storage.Storage) → None

property current_subgraph_state

embeddingsOffDevice(self: marius._storage.GraphModelStorage) → bool

property full_graph_evaluation

getNumActiveEdges(self: marius._storage.GraphModelStorage) → int

getNumActiveNodes(self: marius._storage.GraphModelStorage) → int

getNumEdges(self: marius._storage.GraphModelStorage) → int

getNumNodes(self: marius._storage.GraphModelStorage) → int

getNumNodesInMemory(self: marius._storage.GraphModelStorage) → int

getNumPartitions(self: marius._storage.GraphModelStorage) → int

hasSwap(self: marius._storage.GraphModelStorage) → bool

load(self: marius._storage.GraphModelStorage) → None

property next_subgraph_state

performSwap(self: marius._storage.GraphModelStorage) → None

setFilterEdges(self: marius._storage.GraphModelStorage, arg0: List[marius._storage.Storage]) → None

setTestSet(self: marius._storage.GraphModelStorage) → None

setTrainSet(self: marius._storage.GraphModelStorage) → None

setValidationSet(self: marius._storage.GraphModelStorage) → None

set_edge_storage(self: marius._storage.GraphModelStorage, edge_storage: marius._storage.Storage) → None

set_node_storage(self: marius._storage.GraphModelStorage, node_storage: marius._storage.Storage) → None

shuffleEdges(self: marius._storage.GraphModelStorage) → None

sort_all_edges(self: marius._storage.GraphModelStorage) → None

property storage_ptrs

unload(self: marius._storage.GraphModelStorage, write: bool) → None

update_subgraph(self: marius._storage.GraphModelStorage) → None

useInMemorySubGraph(self: marius._storage.GraphModelStorage) → bool