O'Reilly logo

IPython Interactive Computing and Visualization Cookbook by Cyrille Rossant

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

Using support vector machines for classification tasks

In this recipe, we introduce support vector machines, or SVMs. These powerful models can be used for classification and regression. Here, we illustrate how to use linear and nonlinear SVMs on a simple classification task.

How to do it...

  1. Let's import the packages:
    In [1]: import numpy as np
            import pandas as pd
            import sklearn
            import sklearn.datasets as ds
            import sklearn.cross_validation as cv
            import sklearn.grid_search as gs
            import sklearn.svm as svm
            import matplotlib.pyplot as plt
            %matplotlib inline
  2. We generate 2D points and assign a binary label according to a linear operation on the coordinates:
    In [2]: X = np.random.randn(200, 2)
            y = X[:, 0] + X[:, 1] > 1
  3. We now fit a linear Support Vector Classifier ...

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required