Как назначить cv_results _ ['params'] из GridSearchCV - PullRequest
0 голосов
/ 17 марта 2020

Я хочу создать фрейм данных с помощью cv_results, но GridSearchCV возвращает список обратно. Не знаю, как назначить mat для ind для создания кадра данных.

Вот пример

from sklearn.feature_selection import SelectKBest, f_classif
from sklearn import datasets

diabetes_X, diabetes_y = datasets.load_diabetes(return_X_y=True)

# Split the data into training/testing sets
X_train = diabetes_X[:-20]
X_test = diabetes_X[-20:]

# Split the targets into training/testing sets
y_train = diabetes_y[:-20]
y_test = diabetes_y[-20:]

pipeline = Pipeline([
                     ('kbest', SelectKBest(f_classif)),
                     ('regressor', KNeighborsRegressor())
                    ])

parameters = {'kbest__k':  list(range(1, X_train.shape[1]+1)),
              'regressor__n_neighbors': list(range(1,21))}

grid = GridSearchCV(pipeline, parameters)
grid.fit(X_train, y_train)


mat = grid.cv_results_['mean_test_score']
ind = grid.cv_results_['params']

Вывод params выглядит следующим образом: enter image description here

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...