Я хочу создать фрейм данных с помощью 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
выглядит следующим образом: