noether.data.pipeline.sample_processors.position_normalization

Classes

PositionNormalizationSampleProcessor

Pre-processes data on a sample-level to normalize positions.

Module Contents

class noether.data.pipeline.sample_processors.position_normalization.PositionNormalizationSampleProcessor(items, raw_pos_min, raw_pos_max, scale=1000)

Bases: noether.data.pipeline.sample_processor.SampleProcessor

Pre-processes data on a sample-level to normalize positions.

Should only be used when multiple items should be normalized with the same normalization. If only one item should be normalized, consider using the preprocessor PositionNormalizer instead.

Parameters:

  • items (set[str]) – The position items to normalize. I.e., keys of the input_sample dictionary that should be normalized.

  • raw_pos_min (collections.abc.Sequence[float]) – The minimum position in the source domain.

  • raw_pos_max (collections.abc.Sequence[float]) – The maximum position in the source domain.

  • scale (int | float) – The maximum value of the position. Defaults to 1000.

items
scale = 1000
raw_pos_min_tensor
raw_pos_max_tensor
raw_size