load_iris

sklearn.datasets.load_iris(*, return_X_y=False, as_frame=False)[исходный код]

Загрузить и вернуть набор данных об ирисах (классификация).

Набор данных об ирисах — классический и очень простой набор данных для многоклассовой классификации.

Классы	3
Образцов на класс	50
Всего образцов	150
Размерность	4
Признаки	вещественные, положительные

Подробнее см. в Руководстве пользователя.

Изменено в версии 0.20: Исправлены две ошибочные точки данных в соответствии со статьёй Фишера. Новая версия совпадает с версией в R, но не с версией в репозитории UCI Machine Learning Repository.

Параметры:

return_X_ybool, default=False: Если True, возвращает (data, target) вместо объекта Bunch. Подробнее об объектах data и target см. ниже.

Добавлено в версии 0.18.
as_framebool, default=False: Если True, данные представлены в виде pandas DataFrame со столбцами соответствующих типов (числовыми). Целевая переменная представлена в виде pandas DataFrame или Series в зависимости от количества целевых столбцов. Если return_X_y имеет значение True, то (data, target) будут объектами pandas DataFrame или Series, как описано ниже.

Добавлено в версии 0.23.

Возвращает:

dataBunch

Объект, подобный словарю, со следующими атрибутами.

data{ndarray, dataframe} формы (150, 4): Матрица данных. Если as_frame=True, data будет объектом pandas DataFrame.
target: {ndarray, Series} формы (150,): Целевая переменная классификации. Если as_frame=True, target будет объектом pandas Series.
feature_names: list: Названия столбцов набора данных.
target_names: ndarray формы (3, ): Названия целевых классов.
frame: DataFrame формы (150, 5): Присутствует только при as_frame=True. DataFrame с data и target.

Добавлено в версии 0.23.
DESCR: str: Полное описание набора данных.
filename: str: Путь к расположению данных.

Добавлено в версии 0.20.

(data, target)кортеж, если return_X_y имеет значение True

Кортеж из двух массивов ndarray. Первый содержит двумерный массив формы (150, 4), каждая строка которого соответствует одному образцу, а столбцы — признакам. Второй массив ndarray формы (150,) содержит целевые значения для образцов. Если as_frame=True, оба массива являются объектами pandas: X — DataFrame, а y — Series.

Добавлено в версии 0.18.

Примеры

Предположим, вас интересуют образцы 10, 25 и 50 и вы хотите узнать названия их классов.

>>> from sklearn.datasets import load_iris
>>> data = load_iris()
>>> samples = [10, 25, 50]
>>> data.target[samples]
array([0, 0, 1])
>>> data.target_names[data.target[samples]]
array(['setosa', 'setosa', 'versicolor'], dtype='<U10')

Более подробный пример работы с набором данных об ирисах см. в разделе Анализ главных компонент (PCA) на наборе данных об ирисах.