Function get_partition_data

alphapy/plots.py:102–135

r"""Get the X, y pair for a given model and partition Parameters ---------- model : alphapy.Model The model object with partition data. partition : alphapy.Partition Reference to the dataset. Returns ------- X : numpy array The feature matrix.

(model, partition)

#

def get_partition_data(model, partition):
    r"""Get the X, y pair for a given model and partition

    Parameters
    ----------
    model : alphapy.Model
        The model object with partition data.
    partition : alphapy.Partition
        Reference to the dataset.

    Returns
    -------
    X : numpy array
        The feature matrix.
    y : numpy array
        The target vector.

    Raises
    ------
    TypeError
        Partition must be train or test.

    """

    if partition == Partition.train:
        X = model.X_train
        y = model.y_train
    elif partition == Partition.test:
        X = model.X_test
        y = model.y_test
    else:
        raise TypeError('Partition must be train or test')

    return X, y


#