RuntimeError: Классификатор не предоставляет атрибуты "coef_" или "feature_importances_" - PullRequest
0 голосов
/ 16 октября 2018

В моем коде выдает ошибку во время выполнения.Здесь я пытаюсь приспособить RFE для данных регрессии.

from sklearn.feature_selection import RFE
from sklearn.svm import SVR           
from sklearn.feature_selection import SelectKBest
from sklearn.preprocessing import *

scaler = StandardScaler().fit(trainFeatures)
xscaled = scaler.transform(trainFeatures)
estimator = SVR()
selector = RFE(estimator, dimension, step=1)
selector = selector.fit(xscaled, trainOutput.ravel())
selectedFeatures = selector.transform(xscaled)

1 Ответ

0 голосов
/ 26 ноября 2018

Согласно этой ссылке , RFE работает только с SVR, когда ядро ​​линейно.

По умолчанию это "rbf", так что:

estimator = SVR(kernel="linear")
...