goregression

README

goregression

The goal is to build a regression nueral net on top of gonum.org/v1/gonum/mat

Documentation

Index

• Variables
• type Activation
  • func Linear(slope float64) Activation
  • func Scale(General Activation, factor float64) Activation
  • func Strech(General Activation, factor float64) Activation
• type Job
• type Model
  • func NewModel(source *rand.Rand, Internal Activation, Output Activation, layers ...int) *Model
  • func (m *Model) Clone() *Model
  • func (m Model) InputSize() int
  • func (m Model) OutputSize() int
  • func (m Model) Predict(start mat.Vector) mat.Vector
  • func (m Model) String() string
• type TrainingContext
  • func (tc *TrainingContext) Train(trainingSet [][]mat.Vector, iterations int, lrate float64, ...)
  • func (tc *TrainingContext) TrainChunked(trainingSet [][]mat.Vector, iterations int, workers int, chunksize int, ...)
• type Workers
  • func (w *Workers) Go(j Job)
  • func (w *Workers) Start(size int)
  • func (w *Workers) Stop()

Variables

var BiLn = Activation{
	Activate: func(f float64) float64 {
		if f == 0 {
			return 0
		}
		if f > 0 {
			return math.Log(1 + f)
		}
		return -math.Log(1 - f)
	},
	Derivative: func(f float64) float64 {
		if f == 0 {
			return 1
		}
		if f > 0 {
			return 1 / (1 + f)
		}
		return 1 / (1 - f)
	},
}

var ReLU = Activation{
	Activate: func(f float64) float64 {
		return max(0, f)
	},
	Derivative: func(f float64) float64 {
		if f > 0 {
			return 1
		}
		return 0
	},
}

var Sigmoid = Activation{
	Activate: sigmoid,
	Derivative: func(f float64) float64 {
		s := sigmoid(f)
		return s * (1 - s)
	},
}

var Tanh = Activation{
	Activate: math.Tanh,
	Derivative: func(f float64) float64 {
		t := math.Tanh(f)
		return 1 - t*t
	},
}

Types

type Activation

type Activation struct {
	Activate   func(float64) float64
	Derivative func(float64) float64
}

func Linear

func Linear(slope float64) Activation

func Scale

func Scale(General Activation, factor float64) Activation

func Strech

func Strech(General Activation, factor float64) Activation

type Job

type Job func()

type Model

type Model struct {
	Weights  []mat.Mutable
	Internal Activation
	Output   Activation
}

func NewModel

func NewModel(source *rand.Rand, Internal Activation, Output Activation, layers ...int) *Model

func (*Model) Clone

func (m *Model) Clone() *Model

func (Model) InputSize

func (m Model) InputSize() int

func (Model) OutputSize

func (m Model) OutputSize() int

func (Model) Predict

func (m Model) Predict(start mat.Vector) mat.Vector

func (Model) String

func (m Model) String() string

type TrainingContext

type TrainingContext struct {
	*Model
	GeneratedNodes []*mat.VecDense
	PreNormalized  []*mat.VecDense
}

func (*TrainingContext) Train

func (tc *TrainingContext) Train(trainingSet [][]mat.Vector, iterations int, lrate float64, debug func(epoch int, err float64))

func (*TrainingContext) TrainChunked

func (tc *TrainingContext) TrainChunked(trainingSet [][]mat.Vector, iterations int, workers int, chunksize int, lrate float64, debug func(epoch int, current *Model))

type Workers

type Workers struct {
	sync.WaitGroup
	// contains filtered or unexported fields
}

func (*Workers) Go

func (w *Workers) Go(j Job)

func (*Workers) Start

func (w *Workers) Start(size int)

func (*Workers) Stop

func (w *Workers) Stop()