MariusGraph

class marius.data.MariusGraph

__init__(self: marius._data.MariusGraph) → None

__init__(self: marius._data.MariusGraph, src_sorted_edges: torch.Tensor, dst_sorted_edges: torch.Tensor, num_nodes_in_memory: int) → None

getEdges(self: marius._data.MariusGraph, incoming: bool = True) → torch.Tensor

getNeighborOffsets(self: marius._data.MariusGraph, incoming: bool = True) → torch.Tensor

getNeighborsForNodeIds(self: marius._data.MariusGraph, node_ids: torch.Tensor, incoming: bool, neighbor_sampling_layer: marius._config.NeighborSamplingLayer, max_neighbors_size: int, rate: float)

getNumNeighbors(self: marius._data.MariusGraph, incoming: bool = True) → torch.Tensor

getRelationIDs(self: marius._data.MariusGraph, incoming: bool = True) → torch.Tensor

to(self: marius._data.MariusGraph, device: torch.device) → None

property active_in_memory_subgraph

clear(self: marius._data.MariusGraph) → None

property dst_sorted_edges

property in_num_neighbors

property in_offsets

property in_sorted_uniques

property max_in_num_neighbors_

property max_out_num_neighbors_

property node_ids

property num_nodes_in_memory

property out_num_neighbors

property out_offsets

property out_sorted_uniques

property src_sorted_edges