Я не могу использовать sklearn.model_selection.GridSearchCV с tf.estimator.BoostedTreesClassifier - PullRequest
Новая
       16

Я не могу использовать sklearn.model_selection.GridSearchCV с tf.estimator.BoostedTreesClassifier

0 голосов
/ 11 января 2020

Это код для «модели»: 

n_batches = 1000
model = tf.estimator.BoostedTreesClassifier(feat_cols,
                                          n_batches_per_layer=n_batches)

при применении оптимизации гиперпараметров sklearn, например: 

parameters = {'n_batches_per_layer':[100,1000,5000],'n_trees':[10,100,500],'max_depth':[2,4,6],'learning_rate':[0.01,0.1,0.5]}
grid_search = GridSearchCV(estimator=model,param_grid=parameters,scoring = 'accuracy',
                           cv = 10,
                           n_jobs = -1)
grid_search = grid_search.fit(X_train, y_train)

Я получаю эту ошибку

Оценщик должен быть оценщиком, реализующим метод 'fit', был передан

...