generic

type BasicHmm interface {
	BasicDistribution
	String() string
}

type BasicMixture interface {
	BasicDistribution
	String() string
}

type BaumWelchHook struct {
	Value func(hmm BasicHmm, i int, likelihood, epsilon float64)
}

type BaumWelchOptimizeEmissions struct {
	Value bool
}

type BaumWelchOptimizeTransitions struct {
	Value bool
}

type BaumWelchTmp struct {
	// contains filtered or unexported fields
}

type ChmmTransitionMatrix struct {
	Matrix
	// contains filtered or unexported fields
}

type EmHook struct {
	Value func(mixture BasicMixture, i int, likelihood, epsilon float64)
}

type EmOptimizeEmissions struct {
	Value bool
}

type EmOptimizeWeights struct {
	Value bool
}

type EmTmp struct {
	// contains filtered or unexported fields
}

type EqualityConstraint [][2]int

type HhmmTransitionMatrix struct {
	Matrix
	Tree HmmNode
}

type Hmm struct {
	Pi ProbabilityVector // initial probability vector
	Tr TransitionMatrix  // transition matrix
	Tf TransitionMatrix  // transition matrix for last transition
	// state equivalence classes define which states share the same
	// emission distribution
	StateMap []int
	// number of states
	M int
	// number of emission distributions
	N int
	// contains filtered or unexported fields
}

type HmmDataRecord interface {
	// map index of the kth observation
	MapIndex(k int) int
	// get number of observations in this record
	GetN() int
	// evaluate component c at position k
	LogPdf(r Scalar, c, k int) error
}

type HmmDataSet interface {
	GetRecord(i int) HmmDataRecord
	// number of mapped observations
	GetNMapped() int
	// number of records in the data set
	GetNRecords() int
	// total number of observations
	GetN() int
}

type HmmNode struct {
	Children []HmmNode
	States   [2]int
}

type HmmProbabilityVector struct {
	Vector
	// contains filtered or unexported fields
}

type HmmTransitionMatrix struct {
	Matrix
	// contains filtered or unexported fields
}

type Mixture struct {
	LogWeights Vector
	// contains filtered or unexported fields
}

type MixtureDataRecord interface {
	// evaluate component c
	LogPdf(r Scalar, c int) error
}

type MixtureDataSet interface {
	// evaluate component c on observation i
	LogPdf(r Scalar, c, i int) error
	// total number of observations
	GetCounts() []int
	// total number of observations
	GetN() int
}

type ProbabilityVector interface {
	Vector
	GetVector() Vector
	Normalize() error
	CloneProbabilityVector() ProbabilityVector
}

type TransitionMatrix interface {
	Matrix
	GetMatrix() Matrix
	Normalize() error
	CloneTransitionMatrix() TransitionMatrix
}