Documentation
¶
Index ¶
- type Model
- func (m *Model) Layers() []layer.Layer
- func (m *Model) Load(dir string) error
- func (m *Model) Loss(input, targets mat.Matrix) float64
- func (m *Model) ParamCount() uint64
- func (m *Model) Predict(input mat.Matrix) mat.Matrix
- func (m *Model) Print()
- func (m *Model) ReadFrom(r io.Reader) (int64, error)
- func (m *Model) Save(dir string) error
- func (m *Model) SetLr(lr float64)
- func (m *Model) SetName(name string)
- func (m *Model) Train(input mat.Matrix, targets mat.Matrix)
- func (m *Model) WriteTo(w io.Writer) (int64, error)
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Model ¶
type Model struct {
// contains filtered or unexported fields
}
func (*Model) ParamCount ¶
Source Files
Click to show internal directories.
Click to hide internal directories.