GridSearchCV: получить атрибуты модели - PullRequest
3 голосов
/ 01 ноября 2019

Я использую GridSearchCV, чтобы соответствовать SVM, и я хочу знать количество векторов поддержки для всех установленных моделей. Пока я могу получить доступ только к этому атрибуту SVM для лучшей модели.

Пример игрушки:

import numpy as np
from sklearn.svm import SVC
from sklearn.model_selection import GridSearchCV
X = np.array([[-1, -1], [-2, -1], [1, 1], [2, 1]])
y = np.array([1, 1, 2, 2])
clf = SVC()
params = {'C': [0.01, 0.1, 1]}
search = GridSearchCV(estimator=clf, cv=2, param_grid=params, return_train_score=True)
search.fit(X, y);

с количеством векторов поддержки для лучшей модели:

search.best_estimator_.n_support_

Как получить n_support_ для всех моделей? Так же, как мы получаем ошибку поезда / теста отдельно для каждого параметра C.

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