crf

Documentation

Index

• func Viterbi(transitionMatrix mat.Matrix, xs []ag.Node) []int
• type Model
  • func New(size int) *Model
  • func (m *Model) Decode(emissionScores []ag.Node) []int
  • func (m *Model) InitProcessor()
  • func (m *Model) NegativeLogLoss(emissionScores []ag.Node, target []int) ag.Node
• type ViterbiStructure
  • func NewViterbiStructure(size int) *ViterbiStructure

Functions

func Viterbi

func Viterbi(transitionMatrix mat.Matrix, xs []ag.Node) []int

Viterbi decodes the xs sequence according to the transitionMatrix.

Types

type Model

type Model struct {
	nn.BaseModel
	Size             int
	TransitionScores nn.Param    `spago:"type:weights"`
	Scores           [][]ag.Node `spago:"scope:processor"`
}

Model contains the serializable parameters.

func New

func New(size int) *Model

New returns a new convolution Model, initialized according to the given configuration.

func (*Model) Decode

func (m *Model) Decode(emissionScores []ag.Node) []int

Decode performs viterbi decoding.

func (*Model) InitProcessor

func (m *Model) InitProcessor()

InitProcessor initializes structures and data useful for the decoding.

func (*Model) NegativeLogLoss

func (m *Model) NegativeLogLoss(emissionScores []ag.Node, target []int) ag.Node

NegativeLogLoss computes the negative log loss with respect to the targets.

type ViterbiStructure

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

ViterbiStructure implements Viterbi decoding.

func NewViterbiStructure

func NewViterbiStructure(size int) *ViterbiStructure

NewViterbiStructure returns a new ViterbiStructure ready to use.