Versions in this module Expand all Collapse all v0 v0.9.1 Apr 20, 2024 v0.9.0 Apr 18, 2024 Changes in this version + const PaddingIndex + func MapInputs[T any](strategy *Strategy, inputs []T) map[string]*ValueMask[T] + func NameForNodeDependentDegree(ruleName, dependentName string) string + type Dataset struct + func (ds *Dataset) Epochs(n int) *Dataset + func (ds *Dataset) Infinite() *Dataset + func (ds *Dataset) Name() string + func (ds *Dataset) Reset() + func (ds *Dataset) Shuffle() *Dataset + func (ds *Dataset) WithReplacement() *Dataset + func (ds *Dataset) Yield() (spec any, inputs, labels []tensor.Tensor, err error) + type EdgeType struct + EdgeTargets []int32 + Name string + SourceNodeType string + Starts []int32 + TargetNodeType string + func (et *EdgeType) EdgeTargetsForSourceIdx(srcIdx int32) []int32 + func (et *EdgeType) NumEdges() int + func (et *EdgeType) NumSourceNodes() int + func (et *EdgeType) NumTargetNodes() int + type Rule struct + ConvKernelScopeName string + Count int + Dependents []*Rule + EdgeType *EdgeType + Name string + NodeSet []int32 + NodeTypeName string + NumNodes int32 + Sampler *Sampler + Shape shapes.Shape + SourceRule *Rule + Strategy *Strategy + UpdateKernelScopeName string + func (r *Rule) FromEdges(name, edgeTypeName string, count int) *Rule + func (r *Rule) IdentitySubRule(name string) *Rule + func (r *Rule) IsIdentitySubRule() bool + func (r *Rule) IsNode() bool + func (r *Rule) String() string + func (r *Rule) WithKernelScopeName(name string) *Rule + type Sampler struct + EdgeTypes map[string]*EdgeType + Frozen bool + NodeTypesToCount map[string]int32 + func Load(filePath string) (s *Sampler, err error) + func New() *Sampler + func (s *Sampler) AddEdgeType(name, sourceNodeType, targetNodeType string, edges tensor.Tensor, reverse bool) + func (s *Sampler) AddNodeType(name string, count int) + func (s *Sampler) NewStrategy() *Strategy + func (s *Sampler) Save(filePath string) (err error) + func (s *Sampler) String() string + type Strategy struct + KeepDegrees bool + Rules map[string]*Rule + Sampler *Sampler + Seeds []*Rule + func (strategy *Strategy) NewDataset(name string) *Dataset + func (strategy *Strategy) Nodes(name, nodeTypeName string, count int) *Rule + func (strategy *Strategy) NodesFromSet(name, nodeTypeName string, count int, nodeSet []int32) *Rule + func (strategy *Strategy) String() string + type ValueMask struct + Mask T + Value T