data

Documentation

Index

• func CreateDataSplit(trainSize, testSize, validationSize, size int) ([]int, []int, []int)
• func DegreeMatrix(adj [][]float64) [][]float64
• func EncodeOneHot(labels []string) []int32
• func IdentityMatrix(n int) [][]float64
• func LoadData(path, dataset string) ([][]float64, [][]float64, []int32)
• func NormalizeAdjacencyMatrix(adj [][]float64) [][]float64
• func Sum(row []float64) float64

Functions

func CreateDataSplit

func CreateDataSplit(trainSize, testSize, validationSize, size int) ([]int, []int, []int)

func DegreeMatrix

func DegreeMatrix(adj [][]float64) [][]float64

func EncodeOneHot

func EncodeOneHot(labels []string) []int32

func IdentityMatrix

func IdentityMatrix(n int) [][]float64

func LoadData

func LoadData(path, dataset string) ([][]float64, [][]float64, []int32)

func NormalizeAdjacencyMatrix

func NormalizeAdjacencyMatrix(adj [][]float64) [][]float64

func Sum

func Sum(row []float64) float64