din

Documentation

Index

• type DinNet
  • func NewDinNet(uProfileDim, uBehaviorSize, uBehaviorDim int, iFeatureDim int, cFeatureDim int) *DinNet
  • func NewDinNetFromJson(data []byte) (din *DinNet, err error)
  • func (din *DinNet) Fwd(xUserProfile, xUbMatrix, xItemFeature, xCtxFeature *G.Node, ...) (err error)
  • func (din *DinNet) Graph() *G.ExprGraph
  • func (din *DinNet) In() G.Nodes
  • func (din *DinNet) Learnable() G.Nodes
  • func (din *DinNet) Marshal() (data []byte, err error)
  • func (din *DinNet) Out() *G.Node
  • func (din *DinNet) SetVM(vm G.VM)
  • func (din *DinNet) Vm() G.VM

Types

type DinNet

type DinNet struct {
	// contains filtered or unexported fields
}

func NewDinNet

func NewDinNet(
	uProfileDim, uBehaviorSize, uBehaviorDim int,
	iFeatureDim int,
	cFeatureDim int,
) *DinNet

func NewDinNetFromJson

func NewDinNetFromJson(data []byte) (din *DinNet, err error)

func (*DinNet) Fwd

func (din *DinNet) Fwd(xUserProfile, xUbMatrix, xItemFeature, xCtxFeature *G.Node, batchSize, uBehaviorSize, uBehaviorDim int) (err error)

Fwd performs the forward pass xUserProfile: [batchSize, userProfileDim] xUbMatrix: [batchSize, uBehaviorSize* uBehaviorDim] xUserBehaviors: [batchSize, uBehaviorSize, uBehaviorDim] xItemFeature: [batchSize, iFeatureDim] xContextFeature: [batchSize, cFeatureDim]

func (*DinNet) Graph

func (din *DinNet) Graph() *G.ExprGraph

func (*DinNet) In

func (din *DinNet) In() G.Nodes

func (*DinNet) Learnable

func (din *DinNet) Learnable() G.Nodes

func (*DinNet) Marshal

func (din *DinNet) Marshal() (data []byte, err error)

func (*DinNet) Out

func (din *DinNet) Out() *G.Node

func (*DinNet) SetVM

func (din *DinNet) SetVM(vm G.VM)

func (*DinNet) Vm

func (din *DinNet) Vm() G.VM