Dask SuccessiveHalvingSearchCV给出了TypeError:不支持的操作数类型*：'；非类型'；和'；int'；

我已经尝试了来自的示例，以获得成功的HalvingSearchCV

from sklearn.datasets import make_circles
import numpy as np
from sklearn.neural_network import MLPClassifier
import dask.array as da
from dask_ml.model_selection import SuccessiveHalvingSearchCV
from distributed import Client
client = Client()

X, y = make_circles(n_samples=30_000, random_state=0, noise=0.09)
model = MLPClassifier()
params = {
"hidden_layer_sizes": [
    (24, ),
    (12, 12),
    (6, 6, 6, 6),
    (4, 4, 4, 4, 4, 4),
    (12, 6, 3, 3),
],
"activation": ["relu", "logistic", "tanh"],
"alpha": np.logspace(-6, -3, num=1000),
"batch_size": [16, 32, 64, 128, 256, 512],}

n_examples = 4 * len(X)
n_params = 8
max_iter = n_params  
chunks = n_examples 
X_train = da.from_array(X, chunks=chunks)
y_train = da.from_array(y, chunks=chunks)

search = SuccessiveHalvingSearchCV(model, params, max_iter=max_iter,)
search.fit(X_train, y_train, classes=np.unique(y))

运行search.fit（）时，方法会给出

TypeError:不支持的操作数类型*：'NoneType'和'int'

error。上述dask文档中的示例适用于HyperbandSearchCV和InverseDecaySearchCV

编辑：找出错误。调用

SuccessiveHalvingSearchCV（）

初始化参数n_initial_参数和n_initial_iter时。 示例：

SuccessiveHalvingSearchCV（模型、参数、最大iter=max\u iter、n\u初始参数=10、n\u初始iter=5）