ISIExtractor

class ts3.models.inductive.ISIExtractor(n_bins=128, t_min=0.001, t_max=3.0, log_bins=True)

Bases: ts3.models.base.Extractor

property name: str

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

encode(regime)

Embeddings and uids for every unit of regime this extractor can represent.

Returns (N, D) and (N,) in matching order. Units the extractor cannot represent may be omitted: the probe join drops anything TS3 does not score, and insists separately that the eval side is complete.

Return type:

tuple[Tensor, ndarray]