Conjuntos de dados integrados

Em breve, você treinará modelos, e este capítulo prepara os dados para eles. O scikit-learn, a biblioteca de ML, vem com conjuntos de dados clássicos, como estas flores de íris:

Uma linha os carrega. from ... import importa uma única função de um módulo, então você a chama pelo nome sem qualificação:

from sklearn.datasets import load_iris

iris = load_iris(as_frame=True)
print(iris.frame.shape)
print(iris.target_names)

Isso imprime:

(150, 5)
['setosa' 'versicolor' 'virginica']

as_frame=True solicita pandas, então iris.frame é um DataFrame e iris.feature_names lista suas medidas. target contém números, necessários para os modelos: iris.target_names[2] nomeia a classe 2.