from sklearn import datasets, svm import numpy as np import matplotlib.pyplot as plt from matplotlib.colors import ListedColormap iris = datasets.load_iris() X = iris.data y = iris.target X = X[:,:2] X = X[y!=2] y = y[y!=2]