markov

type Corpus ¶

type Corpus map[string][]*Fragment

Corpus contains data that makes up the Markov chain.

type Fragment ¶

type Fragment struct {
	Words string
	Refs  []string
}

Fragment contains fragmented data used to form a sentence.

type GenerateOption ¶

type GenerateOption func(*generateOptions)

GenerateOption is used to configure a Markov chain's generation settings.

func WithFilter(fn func(*Result) bool) GenerateOption

func WithMaxTries(mt int) GenerateOption

func WithPRNG(fn func() int) GenerateOption

type Markov ¶

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

Markov represents a Markov chain which can be trained and used to generate sentences.

func New(opts ...Option) *Markov

func (m *Markov) AddData(ctx context.Context, data []string)

func (m *Markov) Generate(ctx context.Context, opts ...GenerateOption) (*Result, error)

type Option ¶

type Option func(*Markov)

Option is used to configure a Markov chain.

func WithStateSize(ss int) Option

func WithStore(s Store) Option

type Result ¶

type Result struct {
	String string
	Score  int
	Refs   []string
	Tries  int
}

Result contains data pertaining to a generation.

type Store ¶

type Store interface {
	StartWords(context.Context) []*Fragment
	SetStartWords(context.Context, []*Fragment)
	EndWords(context.Context) []*Fragment
	SetEndWords(context.Context, []*Fragment)
	Corpus(context.Context) Corpus
	SetCorpus(context.Context, Corpus)

	Import(context.Context, *Structure)
	Export(context.Context) *Structure
}

type Structure ¶

type Structure struct {
	Corpus     Corpus
	StartWords []*Fragment
	EndWords   []*Fragment
}

Structure represents the internal structure of a Markov chain and is used to import & export data.