print 'Number of support vectors: ', svc.n_support_


if __name__ == '__main__':
    data1 = loadmat('ex6/ex6data1.mat')

    X1 = data1['X']
    y1 = data1['y']

    print 'X1:', X1.shape
    print 'y1:', y1.shape

    axes = plotData(X1, y1, 'X1', 'X2', ['y == 1', 'y == 0'])
    plt.show()

    clf = SVM.SVC(C=1.0, kernel='linear')
    clf.fit(X1, y1.ravel())
    plotSVC(clf, X1, y1)
    plt.show()

    clf.set_params(C=100)
    clf.fit(X1, y1.ravel())
    plotSVC(clf, X1, y1)
    plt.show()

    data2 = loadmat('ex6/ex6data2.mat')
    X2 = data2['X']
    y2 = data2['y']
    axes = plotData(X2, y2, 'X1', 'X2', ['y == 1', 'y == 0'])
    plt.show()