tlc.client.torch.metrics.predictor#

A wrapper for PyTorch models enabling embedding extraction.

Module Contents#

Classes#

Class

Description

Predictor

A class which can be used to make predictions on a batch of inputs using a model.

API#

class tlc.client.torch.metrics.predictor.Predictor(model: torch.nn.Module, layers: Sequence[int] | None = None)#

A class which can be used to make predictions on a batch of inputs using a model.