engine

package
v0.1.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: May 9, 2021 License: GPL-3.0 Imports: 6 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Engine

type Engine struct {
	NetworkInput        int
	NetworkLayer        []int
	NetworkOutput       int
	Data                *learn.Set
	WinnerNetwork       *neural.Network
	WinnerEvaluation    evaluation.Evaluation
	Verbose             bool
	Usage               neural.NetworkType
	RegressionThreshold float64
}

Engine contains every necessary for starting the engine

func NewEngine

func NewEngine(usage neural.NetworkType, hiddenLayer []int, data *learn.Set) *Engine

NewEngine creates a new Engine object

func (*Engine) GetWinner

func (e *Engine) GetWinner() (*neural.Network, *evaluation.Evaluation)

GetWinner returns the winner network from training

func (*Engine) SetRegressionThreshold

func (e *Engine) SetRegressionThreshold(threshold float64)

SetRegressionThreshold sets the evaluation threshold for the regression

func (*Engine) SetVerbose

func (e *Engine) SetVerbose(verbose bool)

SetVerbose set verbose mode default = false

func (*Engine) Start

func (e *Engine) Start(criterion neural.Criterion, tries, epochs int, trainingSplit, startLearning, decay float64)

Start takes the paramter to start the engine and run it

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL