GridSearchCV и его важность - PullRequest
0 голосов
/ 18 марта 2019

В gridsearchCV, когда я подхожу как-то так:

forest_reg = RandomForestRegressor()
grid_search = GridSearchCV(forest_reg, param_grid,cv=5,scoring = 'neg_mean_squared_error')
grid_search.fit(X_train,y_train)

и после этого, когда я выполню это,

GridSearch.best_estimator_.feature_importances_ 

это дает массив значений поэтому мой вопрос в том, какие значения GridSearch.best_estimator_.feature_importances_ эта строка возвращает ??

1 Ответ

0 голосов
/ 18 марта 2019

В вашем случае GridSearch.best_estimator_.feature_importances_ возвращает объект RandomForestRegressor.

Следовательно, согласно RandomForestRegressor документации :

feature_importances_: массив формы = [n_features] Вернуть значения функций (чем выше, тем важнее функция).

Другими словами, он возвращает наиболее важные функции в соответствии с вашим обучающим набором X_train. Каждый элемент feature_importances_ соответствует одному элементу X_train (например, первый элемент feature_importances_ относится к первому элементу / столбцу X_train).

Чем выше значение элемента в feature_importances_, тем важнее функция в X_train.

...