#!python3

import matplotlib.pyplot as plt
from matplotlib import style
import numpy as np

style.use('ggplot')

class Support_Vector_Machine:
	def __init__(self, visualization=True):
		self.visualization = visualization
		self.colors = {1:'r', -1:'b'}
		if self.visualization:
			self.fig = plt.figure()
			self.ax = self.fig.add_subplot(1,1,1)

	def fit(self, data):
		pass

	def predict(self, features):
		# sign (x.w+b)
		classfication = np.sign(np.dot(np.array(features), self.w) + self.b)

	return classfication



data_dict = {-1:np.array([[1,7],
						  [2,8],
						  [3,8],]), 
			
			1:np.array([[5,1],
						[6,-1],
						[7,3],])}

