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, )