emmi.pipeline.sample_processors.supernode_sampling¶

Classes¶

SupernodeSamplingSampleProcessor

Randomly samples supernodes from a pointcloud.

Module Contents¶

class emmi.pipeline.sample_processors.supernode_sampling.SupernodeSamplingSampleProcessor(item, num_supernodes, supernode_idx_key='supernode_idx', items_at_supernodes=None, seed=None)¶

Bases: ksuit.data.SampleProcessor

Randomly samples supernodes from a pointcloud.

Parameters:

item (str) – Which pointcloud item is used to sample supernodes.
num_supernodes (int) – How many supernodes to sample.
items_at_supernodes (set[str] | None) – Selects items at the supernodes (e.g., pressure at supernodes). Defaults to None.
seed (int | None) – Random seed for deterministic sampling for evaluation. Default None (i.e., no seed). If not None, requires sample index to be present in batch.
supernode_idx_key (str)

item¶

num_supernodes¶

supernode_idx_key = 'supernode_idx'¶

items_at_supernodes = None¶

seed = None¶