Seleccionemos las características sepal_length y sepal_width para las especies setosa y virginica:
iris = sns.load_dataset("iris")
iris = iris[iris.species.isin(["setosa", "virginica"])]
iris["label"] = iris.species.astype("category").cat.codes
iris = iris[iris.species.isin(["setosa", "virginica"])]
iris["label"] = iris.species.astype("category").cat.codes
y = iris.label.values
X = iris[["sepal_length", "sepal_width"]].values
X = iris[["sepal_length", "sepal_width"]].values
Mostremos los datos en el plano:
