From d37c9c458a8a92e75a174fc2efef523551db71b5 Mon Sep 17 00:00:00 2001 From: Matthias Feurer Date: Tue, 12 Nov 2019 11:17:14 +0100 Subject: [PATCH] add support for MLP HP layer_sizes --- openml/extensions/sklearn/extension.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/openml/extensions/sklearn/extension.py b/openml/extensions/sklearn/extension.py index ca6c77458..b3a194756 100644 --- a/openml/extensions/sklearn/extension.py +++ b/openml/extensions/sklearn/extension.py @@ -1927,9 +1927,10 @@ def _obtain_arff_trace( param_value is None or param_value is np.ma.masked: # basic string values type = 'STRING' - elif isinstance(param_value, list) and \ + elif isinstance(param_value, (list, tuple)) and \ all(isinstance(i, int) for i in param_value): - # list of integers + # list of integers (usually for selecting features) + # hyperparameter layer_sizes of MLPClassifier type = 'STRING' else: raise TypeError('Unsupported param type in param grid: %s' % key)