def test_fetch_keel_banana_dobscv(): """Tests Keel banana dataset with dobscv folds.""" data = fetch('classification', 'banana', nfolds=5, dobscv=True) check(data, (5300, 2), splits=5)
def test_fetch_keel_banana_folds(): """Tests Keel banana dataset with folds.""" data = fetch('classification', 'banana', nfolds=5) check(data, (5300, 2), splits=5)
def test_fetch_keel_banana(): """Tests Keel banana dataset.""" data = fetch('classification', 'banana') check(data, (5300, 2))
def test_fetch_keel_abalone9_18_folds(): """Tests Keel abalone9-18 dataset with folds.""" data = fetch('imbalanced', 'abalone9-18', nfolds=5) check(data, (731, 10), splits=5)
def test_fetch_keel_abalone9_18(): """Tests Keel abalone9-18 dataset.""" data = fetch('imbalanced', 'abalone9-18') check(data, (731, 10))
def test_fetch_keel_abalone9_18_return_X_y(): """Tests Keel abalone9-18 dataset.""" X, y = fetch('imbalanced', 'abalone9-18', return_X_y=True) assert X.shape == (731, 10) assert y.shape == (731, )