from SVM import *

from documento import *

from pattern.web import *

from model import *

import os

# reload(sys) # Reload does the trick!

# # sys.setdefaultencoding('UTF8')

# print sys.getdefaultencoding()

lecturaArchivo('data/documentos.csv','entrenamiento')

X = getDocumentosAtributos('entrenamiento')

Y = getDocumentosClase()

unSVM = SVM(1.0,'poly',.7,.3,X,Y)

unSVM.training()

print "Precision : " ,unSVM.testing()

lecturaArchivo('data/prediccion.csv','predecir')

X = getDocumentosAtributos('predecir')

print unSVM.predecir(X)