FIX remove multilabel indicator from sgd component

automl · Jan 3, 2017 · 73e2863 · 73e2863
1 parent 9a62e98
commit 73e2863
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/autosklearn/pipeline/components/classification/sgd.py b/autosklearn/pipeline/components/classification/sgd.py
@@ -106,7 +106,7 @@ def get_properties(dataset_properties=None):
                 'handles_regression': False,
                 'handles_classification': True,
                 'handles_multiclass': True,
-                'handles_multilabel': True,
+                'handles_multilabel': False,
                 'is_deterministic': True,
                 'input': (DENSE, SPARSE, UNSIGNED_DATA),
                 'output': (PREDICTIONS,)}

diff --git a/test/test_pipeline/test_regression.py b/test/test_pipeline/test_regression.py
@@ -101,7 +101,6 @@ def _test_configurations(self, configurations_space, make_sparse=False,
 
         configurations_space.seed(1)
 
-        print(configurations_space)
         for i in range(10):
             config = configurations_space.sample_configuration()
             config._populate_values()
@@ -125,8 +124,6 @@ def _test_configurations(self, configurations_space, make_sparse=False,
                                 config[restrict_parameter] is not None:
                     config._values[restrict_parameter] = restrict_to
 
-            print(config)
-
             if data is None:
                 X_train, Y_train, X_test, Y_test = get_dataset(
                     dataset='boston', make_sparse=make_sparse, add_NaNs=True)