alibi_detect.models.pytorch.gmm module

class alibi_detect.models.pytorch.gmm.GMMModel(n_components, dim)[source]

Bases: Module

__init__(n_components, dim)[source]

Gaussian Mixture Model (GMM).

Parameters:

  • n_components (int) – The number of mixture components.

  • dim (int) – The dimensionality of the data.

forward(x)[source]

Compute the log-likelihood of the data.

Parameters:

x (Tensor) – Data to score.

Return type:

Tensor