Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Config ¶ added in v0.2.5
type Config struct {
Tries int
Epochs int
TrainingSplit float64
LearningRate float64
Decay float64
RegressionThreshold float64
}
Config has all the learning configurations necessary to learn the netowrk in the engine
func DefaultConfig ¶ added in v0.2.5
func DefaultConfig() *Config
DefaultConfig returns the default config for the engine learner
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
Config *Config
}
Engine contains every necessary for starting the engine
func (*Engine) GetConfig ¶ added in v0.2.5
GetConfig returns the current engine learner configuration
func (*Engine) GetWinner ¶
func (e *Engine) GetWinner() (*neural.Network, *evaluation.Evaluation)
GetWinner returns the winner network from training
func (*Engine) SetConfig ¶ added in v0.2.5
SetConfig sets a new config from outside for the engine learner
func (*Engine) SetRegressionThreshold ¶
SetRegressionThreshold sets the evaluation threshold for the regression
func (*Engine) SetVerbose ¶
SetVerbose set verbose mode default = false
Click to show internal directories.
Click to hide internal directories.