API Reference

Bayesian Models

Gaussian Mixture

GaussianMixture	Probabilistic model assuming that data is generated from a mixture of Gaussians.
GaussianMixtureModel	PyTorch module for a Gaussian mixture model.
GaussianMixtureModelConfig	Configuration class for a Gaussian mixture model.

Markov Chain

MarkovChain	Probabilistic model for observed state transitions.
MarkovChainModel	PyTorch module for a Markov chain.
MarkovChainModelConfig	Configuration class for a Markov chain model.

Clustering Models

K-Means

KMeans	Model for clustering data into a predefined number of clusters.
KMeansModel	PyTorch module for the K-Means model.
KMeansModelConfig	Configuration class for a K-Means model.

Utility Types

SequenceData	Data that may be passed to estimators expecting 1-D sequences.
CovarianceType	The type of covariance to use for a set of multivariate Normal distributions.
GaussianMixtureInitStrategy	Strategy for initializing the parameters of a Gaussian mixture model.
KMeansInitStrategy	Strategy for initializing KMeans centroids.