symbol.go

package nn

import (
	"go4ml.xyz/iokit"
	"go4ml.xyz/nn/mx"
	"gopkg.in/yaml.v3"
	"io"
	"sync"
)

var symbolMu = sync.Mutex{}
var _symbolId = 0

func NextSymbolId() int {
	_symbolId++
	return _symbolId
}

func resetSymbolId(first int) {
	_symbolId = first
}

func SaveSymbol(inputdim mx.Dimension, sym *mx.Symbol, output iokit.Output) (err error) {
	var wr iokit.Whole
	if wr, err = output.Create(); err != nil {
		return
	}
	defer wr.End()
	enc := yaml.NewEncoder(wr)
	x := struct {
		Input    mx.Dimension `yaml:"input"`
		Symbolic *mx.Symbol   `yaml:"symbolic"`
	}{inputdim, sym}
	if err = enc.Encode(x); err != nil {
		return
	}
	if err = enc.Close(); err != nil {
		return
	}
	return wr.Commit()
}

func (network *Network) SaveSymbol(output iokit.Output) (err error) {
	return SaveSymbol(network.inputdim, network.symbolic, output)
}

func LoadSymbol(input iokit.Input) (symbolic *mx.Symbol, inputdim mx.Dimension, err error) {
	var rd io.ReadCloser
	if rd, err = input.Open(); err != nil {
		return
	}
	defer rd.Close()
	dec := yaml.NewDecoder(rd)
	x := struct {
		Input    mx.Dimension `yaml:"input"`
		Symbolic *mx.Symbol   `yaml:"symbolic"`
	}{}
	if err = dec.Decode(&x); err != nil {
		return
	}
	return x.Symbolic, x.Input, nil
}