decoder

type BeamPQ []*Hypothesis

type ExpectedMatchingBeamDecoder struct {
	// Expected sequence of words/tokens
	ExpectedSequence []int

	// Vocabulary size of model
	VocabSize int

	// Size of beam
	BeamWidth int

	// CTC blank token ID
	BlankID int

	// Weights for scoring
	ExpectedMatchBonus float64 // Bonus for matching expected token
	InsertionPenalty   float64 // Penalty for inserting a token not in expected sequence
	DeletionPenalty    float64 // Penalty for skipping an expected token

	// Maximum lookahead in expected sequence
	MaxLookahead int
}

type Hypothesis struct {
	// Sequence of tokens/words decoded so far
	Tokens []int

	// Position in the expected output sequence
	ExpectedPos int

	// Cumulative log probability score
	Score float64

	// Position in acoustic frames
	TimeStep int
	// contains filtered or unexported fields
}