Skip to content

Commit 322a3a1

Browse files
committed
FIX checking for SPARSE and itertaive fit; None if test can't run
1 parent ba0f78f commit 322a3a1

File tree

10 files changed

+46
-42
lines changed

10 files changed

+46
-42
lines changed

test/test_pipeline/components/regression/test_adaboost.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@ class AdaBoostComponentTest(BaseRegressionComponentTest):
1111

1212
res = dict()
1313
res["default_boston"] = 0.60525743737887405
14-
res["default_boston_iterative"] = -1
14+
res["default_boston_iterative"] = None
1515
res["default_boston_sparse"] = 0.22111559712318207
16-
res["default_boston_iterative_sparse"] = 0.0
16+
res["default_boston_iterative_sparse"] = None
1717
res["default_diabetes"] = 0.25129853514492517
18-
res["default_diabetes_iterative"] = -1
18+
res["default_diabetes_iterative"] = None
1919
res["default_diabetes_sparse"] = 0.090755670764629537
20-
res["default_diabetes_iterative_sparse"] = 0.0
20+
res["default_diabetes_iterative_sparse"] = None
2121

2222
sk_mod = sklearn.ensemble.AdaBoostRegressor
2323

test/test_pipeline/components/regression/test_ard_regression.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@ class ARDRegressionComponentTest(BaseRegressionComponentTest):
1111

1212
res = dict()
1313
res["default_boston"] = 0.70316694175513961
14-
res["default_boston_iterative"] = -1
15-
res["default_boston_sparse"] = -1
16-
res["default_boston_iterative_sparse"] = 0.0
14+
res["default_boston_iterative"] = None
15+
res["default_boston_sparse"] = None
16+
res["default_boston_iterative_sparse"] = None
1717
res["default_diabetes"] = 0.4172236487551515
18-
res["default_diabetes_iterative"] = -1
19-
res["default_diabetes_sparse"] = -1
20-
res["default_diabetes_iterative_sparse"] = 0.0
18+
res["default_diabetes_iterative"] = None
19+
res["default_diabetes_sparse"] = None
20+
res["default_diabetes_iterative_sparse"] = None
2121

2222
sk_mod = sklearn.linear_model.ARDRegression
2323

test/test_pipeline/components/regression/test_base.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
from autosklearn.pipeline.constants import *
77

88
import sklearn.metrics
9-
import numpy as np
109

1110

1211
class BaseRegressionComponentTest(unittest.TestCase):
@@ -59,6 +58,8 @@ def test_default_boston_iterative_fit(self):
5958
def test_default_boston_iterative_sparse_fit(self):
6059
if not hasattr(self.module, 'iterative_fit'):
6160
return
61+
if SPARSE not in self.module.get_properties()["input"]:
62+
return
6263

6364
for i in range(2):
6465
predictions, targets = \
@@ -115,6 +116,8 @@ def test_default_diabetes_iterative_fit(self):
115116
def test_default_diabetes_iterative_sparse_fit(self):
116117
if not hasattr(self.module, 'iterative_fit'):
117118
return
119+
if SPARSE not in self.module.get_properties()["input"]:
120+
return
118121

119122
for i in range(2):
120123
predictions, targets = \

test/test_pipeline/components/regression/test_decision_tree.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@ class DecisionTreeComponentTest(BaseRegressionComponentTest):
1111

1212
res = dict()
1313
res["default_boston"] = 0.35616796434879905
14-
res["default_boston_iterative"] = -1
14+
res["default_boston_iterative"] = None
1515
res["default_boston_sparse"] = 0.18031669797027394
16-
res["default_boston_iterative_sparse"] = 0.0
16+
res["default_boston_iterative_sparse"] = None
1717
res["default_diabetes"] = 0.1564592449511697
18-
res["default_diabetes_iterative"] = -1
18+
res["default_diabetes_iterative"] = None
1919
res["default_diabetes_sparse"] = -0.020818312539637507
20-
res["default_diabetes_iterative_sparse"] = 0.0
20+
res["default_diabetes_iterative_sparse"] = None
2121

2222
sk_mod = sklearn.tree.DecisionTreeRegressor
2323

test/test_pipeline/components/regression/test_gaussian_process.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,13 @@ class GaussianProcessComponentTest(BaseRegressionComponentTest):
1313
res = dict()
1414
res["default_boston_le_ge"] = [0.6, 0.4]
1515
res["default_boston_places"] = 1
16-
res["default_boston_iterative"] = -1
17-
res["default_boston_sparse"] = -1
18-
res["default_boston_iterative_sparse"] = 0.0
16+
res["default_boston_iterative"] = None
17+
res["default_boston_sparse"] = None
18+
res["default_boston_iterative_sparse"] = None
1919
res["default_diabetes"] = -7.4131230585194885e-06
20-
res["default_diabetes_iterative"] = -1
21-
res["default_diabetes_sparse"] = -1
22-
res["default_diabetes_iterative_sparse"] = 0.0
20+
res["default_diabetes_iterative"] = None
21+
res["default_diabetes_sparse"] = None
22+
res["default_diabetes_iterative_sparse"] = None
2323

2424
sk_mod = sklearn.gaussian_process.GaussianProcessRegressor
2525

test/test_pipeline/components/regression/test_gradient_boosting.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import sklearn.ensemble
22

3-
from autosklearn.pipeline.components.regression.gradient_boosting import GradientBoosting
3+
from autosklearn.pipeline.components.regression.gradient_boosting import \
4+
GradientBoosting
45

56
from .test_base import BaseRegressionComponentTest
67

@@ -12,12 +13,12 @@ class GradientBoostingComponentTest(BaseRegressionComponentTest):
1213
res = dict()
1314
res["default_boston"] = 0.83961954550470863
1415
res["default_boston_iterative"] = 0.83961954550470863
15-
res["default_boston_sparse"] = 0.0
16-
res["default_boston_iterative_sparse"] = 0.0
16+
res["default_boston_sparse"] = None
17+
res["default_boston_iterative_sparse"] = None
1718
res["default_diabetes"] = 0.37192663934006487
1819
res["default_diabetes_iterative"] = 0.37192663934006487
19-
res["default_diabetes_sparse"] = 0.0
20-
res["default_diabetes_iterative_sparse"] = 0.0
20+
res["default_diabetes_sparse"] = None
21+
res["default_diabetes_iterative_sparse"] = None
2122

2223
sk_mod = sklearn.ensemble.GradientBoostingRegressor
2324

test/test_pipeline/components/regression/test_k_nearest_neighbors.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,13 @@ class KNearestNeighborsComponentTest(BaseRegressionComponentTest):
1111

1212
res = dict()
1313
res["default_boston"] = 0.18393287980040374
14-
res["default_boston_iterative"] = 0.0
14+
res["default_boston_iterative"] = None
1515
res["default_boston_sparse"] = -0.23029229186279609
16-
res["default_boston_iterative_sparse"] = 0.0
16+
res["default_boston_iterative_sparse"] = None
1717
res["default_diabetes"] = 0.068600456340847438
18-
res["default_diabetes_iterative"] = 0.0
18+
res["default_diabetes_iterative"] = None
1919
res["default_diabetes_sparse"] = -0.16321841460809972
20-
res["default_diabetes_iterative_sparse"] = 0.0
20+
res["default_diabetes_iterative_sparse"] = None
2121

2222
sk_mod = sklearn.neighbors.KNeighborsRegressor
2323

test/test_pipeline/components/regression/test_liblinear_svr.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,13 @@ class SupportVectorComponentTest(BaseRegressionComponentTest):
1010

1111
res = dict()
1212
res["default_boston"] = 0.6768297818275556
13-
res["default_boston_iterative"] = 0.0
13+
res["default_boston_iterative"] = None
1414
res["default_boston_sparse"] = 0.12626519114138912
15-
res["default_boston_iterative_sparse"] = 0.0
15+
res["default_boston_iterative_sparse"] = None
1616
res["default_diabetes"] = 0.39152218711865661
17-
res["default_diabetes_iterative"] = 0.0
17+
res["default_diabetes_iterative"] = None
1818
res["default_diabetes_sparse"] = 0.18704323088631891
19-
res["default_diabetes_iterative_sparse"] = 0.0
19+
res["default_diabetes_iterative_sparse"] = None
2020

2121
sk_mod = sklearn.svm.LinearSVR
2222

test/test_pipeline/components/regression/test_ridge_regression.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,13 @@ class RidgeComponentTest(BaseRegressionComponentTest):
1010

1111
res = dict()
1212
res["default_boston"] = 0.70337988453496891
13-
res["default_boston_iterative"] = 0.0
13+
res["default_boston_iterative"] = None
1414
res["default_boston_sparse"] = 0.11243478302989141
15-
res["default_boston_iterative_sparse"] = 0.0
15+
res["default_boston_iterative_sparse"] = None
1616
res["default_diabetes"] = 0.32614416980439365
17-
res["default_diabetes_iterative"] = 0.0
17+
res["default_diabetes_iterative"] = None
1818
res["default_diabetes_sparse"] = 0.12989713186102791
19-
res["default_diabetes_iterative_sparse"] = 0.0
19+
res["default_diabetes_iterative_sparse"] = None
2020

2121
sk_mod = sklearn.linear_model.Ridge
2222

test/test_pipeline/components/regression/test_support_vector_regression.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,13 @@ class SupportVectorComponentTest(BaseRegressionComponentTest):
99

1010
res = dict()
1111
res["default_boston"] = -0.030006883949312613
12-
res["default_boston_iterative"] = 0.0
12+
res["default_boston_iterative"] = None
1313
res["default_boston_sparse"] = -0.062749211736050192
14-
res["default_boston_iterative_sparse"] = 0.0
14+
res["default_boston_iterative_sparse"] = None
1515
res["default_diabetes"] = 0.12849591861430087
16-
res["default_diabetes_iterative"] = 0.0
16+
res["default_diabetes_iterative"] = None
1717
res["default_diabetes_sparse"] = 0.0098877566961463881
18-
res["default_diabetes_iterative_sparse"] = 0.0
18+
res["default_diabetes_iterative_sparse"] = None
1919

2020
sk_mod = sklearn.svm.SVR
2121

0 commit comments

Comments
 (0)