Esempio n. 1
0
func (l *lstmGate) Serialize() ([]byte, error) {
	slist := []serializer.Serializer{l.Dense, l.Activation}
	if l.Peephole != nil {
		data, err := json.Marshal(l.Peephole)
		if err != nil {
			return nil, err
		}
		slist = append(slist, serializer.Bytes(data))
	}
	return serializer.SerializeSlice(slist)
}
Esempio n. 2
0
// Serialize serializes the block.
func (n *NetworkBlock) Serialize() ([]byte, error) {
	initData, err := json.Marshal(n.batcherBlock.Start)
	if err != nil {
		return nil, err
	}
	serializers := []serializer.Serializer{
		serializer.Int(n.batcherBlock.StateSize),
		n.network,
		serializer.Bytes(initData),
	}
	return serializer.SerializeSlice(serializers)
}
Esempio n. 3
0
// Serialize serializes the block.
func (g *GRU) Serialize() ([]byte, error) {
	initData, err := json.Marshal(g.initState)
	if err != nil {
		return nil, err
	}
	slist := []serializer.Serializer{
		serializer.Int(g.hiddenSize),
		g.inputValue,
		g.resetGate,
		g.updateGate,
		serializer.Bytes(initData),
	}
	return serializer.SerializeSlice(slist)
}
Esempio n. 4
0
// Serialize serializes the LSTM.
func (l *LSTM) Serialize() ([]byte, error) {
	initData, err := json.Marshal(l.initState)
	if err != nil {
		return nil, err
	}
	slist := []serializer.Serializer{
		serializer.Int(l.hiddenSize),
		l.inputValue,
		l.inputGate,
		l.rememberGate,
		l.outputGate,
		serializer.Bytes(initData),
	}
	return serializer.SerializeSlice(slist)
}