class, layer=-1, flatten=False)[source]

Bases: torch.nn.Module

Return type

Tensor, model, device=None, tokenizer=None, max_len=None, batch_size=10000000000, dtype=numpy.float32)[source]

Prediction function used for preprocessing step of drift detector.

  • x (ndarray) – Batch of instances.

  • model (Union[Module, Sequential]) – Model used for preprocessing.

  • device (Optional[device]) – Device type used. The default None tries to use the GPU and falls back on CPU if needed. Can be specified by passing either torch.device(‘cuda’) or torch.device(‘cpu’).

  • tokenizer (Optional[Callable]) – Optional tokenizer for text drift.

  • max_len (Optional[int]) – Optional max token length for text drift.

  • batch_size (int) – Batch size used during prediction.

  • dtype (type) – Model output type, e.g. np.float32 or torch.float32.

Return type

Union[ndarray, Tensor]


Numpy array or torch tensor with predictions.