from sklearn.datasets import make_classification


class SamplingTestData:

    def get_classification_data(self):
        X, Y = make_classification(n_classes=2, class_sep=2,
                               weights=[0.1, 0.9], n_informative=3, n_redundant=1, flip_y=0,
                               n_features=20, n_clusters_per_class=1, n_samples=1000, random_state=10)
        return X, Y