din

package

v0.5.0 Latest Latest Go to latest Published: Nov 6, 2022 License: AGPL-3.0 Imports: 7 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

Repository

github.com/auxten/edgeRec

Links

Open Source Insights

Documentation ¶

Index ¶

type DinNet
- func NewDinNet(uProfileDim, uBehaviorSize, uBehaviorDim int, iFeatureDim int, cFeatureDim int) *DinNet
- func NewDinNetFromJson(data []byte) (din *DinNet, err error)

Constants ¶

This section is empty.

Variables ¶

This section is empty.

Functions ¶

This section is empty.

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 ¶ added in v0.4.0

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

Source Files ¶

View all Source files

din.go

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL