Как получить имя целевой функции из склеарна `Bunch`

1

Можно получить "заголовки столбцов" из sklearn Bunch как Bunch.feature_names. Однако это не дает заголовок столбца для целевой переменной. Как получить заголовок столбца для целевой переменной?

например, для набора данных Iris:

>>> data.feature_names 
['sepal length (cm)', 'sepal width (cm)', 'petal length (cm)', 'petal width(cm)']

В этом духе заголовок столбца для целевой переменной будет чем-то вроде "species" (поскольку data.target_names дает array(['setosa', 'versicolor', 'virginica'], dtype='<U10'). Есть ли у склеарного Bunch средства для получения "заголовка столбца" для цели, например "виды"?

Теги:
scikit-learn

1 ответ

2
Лучший ответ

Точная вещь, которую вы хотите, невозможна. Однако вы можете сделать следующее:

from sklearn.datasets import load_iris

data = load_iris()
print(data.DESCR)

И прочитайте описание

The data set contains 3 classes of 50 instances each, where each class refers to a type of iris plant.

  • 0
    Была надежда, что заголовок столбца существовал, так как у меня есть обширный конвейер, и было бы просто извлечь его и пропустить через него. Я полагаю, что в вашем решении подразумевается, что ответ на вопрос «Есть ли у Bunch средства для получения« заголовка столбца »для цели» - «нет»?
  • 0
    Верный. Нет другого <direct> способа сделать это

Ещё вопросы

Сообщество Overcoder
Наверх
Меню