NEMOExtractor

class ts3.models.inductive.NEMOExtractor(ckpt, batch_size=None)

Bases: ts3.models.base.Extractor

property name: str

Names the embeddings file, so two runs of one model do not overwrite each other.

property run_seed: int | None

The seed of the run behind these embeddings, or None when no run produced them.

setup(data_root, device, logger)

Bind the run’s context. Override to load checkpoints once, not per regime.

Return type:

None

encode(regime)

One embedding per unit, from its waveform and autocorrelogram.

Return type:

tuple[Tensor, ndarray]