def test_classification_dataset(self):
X, y, split = functions.get_sample_dataset(self.dataset_properties)
assert X.shape == (100, 20)
assert y.shape == (100, )
assert len(np.unique(y)) == 2
self.dataset_properties['n_classes'] = 4
self.dataset_properties['n_informative'] = 18
X, y, split = functions.get_sample_dataset(self.dataset_properties)
assert X.shape == (100, 20)
assert y.shape == (100, )
assert len(np.unique(y)) == 4
self.dataset_properties['n_features'] = 100
X, y, split = functions.get_sample_dataset(self.dataset_properties)
assert X.shape == (100, 100)
assert y.shape == (100, )
assert len(np.unique(y)) == 4
self.dataset_properties['n_samples'] = 24
X, y, split = functions.get_sample_dataset(self.dataset_properties)
assert X.shape == (24, 100)
assert y.shape == (24, )
assert len(np.unique(y)) == 4
def test_diabetes(self):
X, y, split = functions.get_sample_dataset({'type': 'diabetes'})
assert X.shape == (442, 10)
assert y.shape == (442, )
def test_boston_housing(self):
X, y, split = functions.get_sample_dataset({'type': 'boston'})
assert X.shape == (506, 13)
assert y.shape == (506, )
def test_breast_cancer_dataset(self):
X, y, split = functions.get_sample_dataset({'type': 'breast_cancer'})
assert X.shape == (569, 30)
assert y.shape == (569, )
def test_mnist_dataset(self):
X, y, split = functions.get_sample_dataset({'type': 'mnist'})
assert X.shape == (1797, 64)
assert y.shape == (1797, )
def test_iris_dataset(self):
X, y, split = functions.get_sample_dataset({'type': 'iris'})
assert X.shape == (150, 4)
assert y.shape == (150, )
