Skip to main content

train_test_split

Function train_test_split 

Source
pub fn train_test_split(
    py: Python<'_>,
    x: PyReadonlyArray2<'_, f64>,
    y: PyReadonlyArray1<'_, f64>,
    test_size: Option<f64>,
    random_state: Option<u64>,
) -> PyResult<(Py<PyArray2<f64>>, Py<PyArray2<f64>>, Py<PyArray1<f64>>, Py<PyArray1<f64>>)>
Expand description

Split arrays into random train and test subsets.

§Notes

The underlying implementation always shuffles before splitting and does not yet support scikit-learn’s train_size or stratify parameters; only test_size and random_state are real, working parameters.