Documentation ¶
Index ¶
- type Layer
- type Model
- func (e *Model) Add(layer Layer)
- func (e Model) Fit(x_train [][][][]float64, y_train [][][][]float64, epochs int, lr float64) Result
- func (e Model) Predict(in [][][]float64) [][][]float64
- func (net *Model) Print(epoch int, epochs int, item int, items int, acc float64, er float64, ...)
- func (e *Model) Save(folder string, file ...string) error
- type Result
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Layer ¶
type Layer interface { Forward([][][]float64) [][][]float64 Backward([][][]float64, float64) [][][]float64 }
Defines a layer with the functions for forward propagation and backward propagation
type Model ¶
Defines a sequential model
Click to show internal directories.
Click to hide internal directories.