youtube

Documentation

Index

• type YoutubeDnn
  • func NewYoutubeDnn(uProfileDim, uBehaviorSize, uBehaviorDim int, iFeatureDim int, cFeatureDim int) (mlp *YoutubeDnn)
  • func NewYoutubeDnnFromJson(data []byte) (mlp *YoutubeDnn, err error)
  • func (mlp *YoutubeDnn) Fwd(xUserProfile, ubMatrix, xItemFeature, xCtxFeature *G.Node, ...) (err error)
  • func (mlp *YoutubeDnn) Graph() *G.ExprGraph
  • func (mlp *YoutubeDnn) In() G.Nodes
  • func (mlp *YoutubeDnn) Learnable() G.Nodes
  • func (mlp *YoutubeDnn) Marshal() (data []byte, err error)
  • func (mlp *YoutubeDnn) Out() *G.Node
  • func (mlp *YoutubeDnn) SetVM(vm G.VM)
  • func (mlp *YoutubeDnn) Vm() G.VM

Types

type YoutubeDnn

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

func NewYoutubeDnn

func NewYoutubeDnn(
	uProfileDim, uBehaviorSize, uBehaviorDim int,
	iFeatureDim int,
	cFeatureDim int,
) (mlp *YoutubeDnn)

func NewYoutubeDnnFromJson

func NewYoutubeDnnFromJson(data []byte) (mlp *YoutubeDnn, err error)

func (*YoutubeDnn) Fwd

func (mlp *YoutubeDnn) Fwd(xUserProfile, ubMatrix, xItemFeature, xCtxFeature *G.Node, batchSize, uBehaviorSize, uBehaviorDim int) (err error)

Fwd ... xUserProfile: [batchSize, userProfileDim] xUbMatrix: [batchSize, uBehaviorSize* uBehaviorDim] xUserBehaviors: [batchSize, uBehaviorSize, uBehaviorDim] xItemFeature: [batchSize, iFeatureDim] xContextFeature: [batchSize, cFeatureDim]

func (*YoutubeDnn) Graph

func (mlp *YoutubeDnn) Graph() *G.ExprGraph

func (*YoutubeDnn) In

func (mlp *YoutubeDnn) In() G.Nodes

func (*YoutubeDnn) Learnable

func (mlp *YoutubeDnn) Learnable() G.Nodes

func (*YoutubeDnn) Marshal

func (mlp *YoutubeDnn) Marshal() (data []byte, err error)

func (*YoutubeDnn) Out

func (mlp *YoutubeDnn) Out() *G.Node

func (*YoutubeDnn) SetVM

func (mlp *YoutubeDnn) SetVM(vm G.VM)

func (*YoutubeDnn) Vm

func (mlp *YoutubeDnn) Vm() G.VM