Table of Contents

Mojo function

train_test_split

def train_test_split(X: Matrix, y: Matrix, *, test_size: Float32 = 0.5, train_size: Float32 = 0) -> Tuple[Matrix, Matrix, Matrix, Matrix]

Split matrices into random train and test subsets.

Args:

  • X (Matrix)
  • y (Matrix)
  • test_size (Float32)
  • train_size (Float32)

Returns:

Tuple

Raises:

def train_test_split(X: Matrix, y: Matrix, *, random_state: Int, test_size: Float32 = 0.5, train_size: Float32 = 0) -> Tuple[Matrix, Matrix, Matrix, Matrix]

Split matrices into random train and test subsets.

Args:

  • X (Matrix)
  • y (Matrix)
  • random_state (Int)
  • test_size (Float32)
  • train_size (Float32)

Returns:

Tuple

Raises:

Edit this page