Documentation ¶
Index ¶
- type YoutubeDnn
- 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
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
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
Click to show internal directories.
Click to hide internal directories.