NuCLRExtractor

class ts3.models.inductive.NuCLRExtractor(ckpt, batch_size=None, num_workers=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 mean embedding per unit, over every window that unit appears in.

Return type:

tuple[Tensor, ndarray]