cross_validation

Documentation

Index

• func TrainTestSplit(size interface{}, randomState interface{}, datasets ...*mat.Dense) ([]*mat.Dense, error)

Functions

func TrainTestSplit

func TrainTestSplit(size interface{}, randomState interface{}, datasets ...*mat.Dense) ([]*mat.Dense, error)

TrainTestSplit splits input DenseMatrix into subsets for testing. The function expects a test size number (int) or percentage (float64), and a random state or nil to get "random" shuffle. It returns a list containing the train-test split and an error status.