Skip to content

Commit 69d443f

Browse files
authored
add support for MLP HP layer_sizes (#879)
1 parent a1e2c34 commit 69d443f

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

openml/extensions/sklearn/extension.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1927,9 +1927,10 @@ def _obtain_arff_trace(
19271927
param_value is None or param_value is np.ma.masked:
19281928
# basic string values
19291929
type = 'STRING'
1930-
elif isinstance(param_value, list) and \
1930+
elif isinstance(param_value, (list, tuple)) and \
19311931
all(isinstance(i, int) for i in param_value):
1932-
# list of integers
1932+
# list of integers (usually for selecting features)
1933+
# hyperparameter layer_sizes of MLPClassifier
19331934
type = 'STRING'
19341935
else:
19351936
raise TypeError('Unsupported param type in param grid: %s' % key)

0 commit comments

Comments
 (0)