from classifier import SVMClassifier
from helper import line, print_line
import numpy

def svm_tiny():
	data = numpy.matrix([[1], [2], [3]])
	labels = [-1.0, 1.0, 1.0]

	svm = SVMClassifier()
	svm.train(data, labels)
	print svm.decision([-2, -1, 0, 1, 1.5, 2, 3, 4])

def svm_line():
	data, labels = line(n=50, noise=0.1, a=1, b=0)
	svm = SVMClassifier()
	svm.train(data, labels, c=1)
	svm.print_2Ddecision('img/svm_line.png')