feyn.datasets

function make_classification

def make_classification(
    n_samples: int = 100,
    n_features: int = 20,
    stratify: bool = False,
    **kwargs
) -> Tuple

Uses sklearn.datasets.make_classification to create a classification problem
and returns train and test DataFrames.
Keyword arguments are parameters in sklearn.datasets.make_classification and
sklearn.model_selection.train_test_split.

Args:
    n_samples (int, optional): The number of samples. Defaults to 100.
    n_features (int, optional): The number of features. Defaults to 20.
    stratify (bool, optional): Stratifies the train, test split by the target variable y. Defaults to False.

Returns:
    train, test: The training and test set of the classification problem