swiggumj / ubc_ai Goto Github PK

UBC AI project (pfd AI part)
by Aaron Berndsen and Weiwei Zhu


TO USE THIS CODE YOU NEED TO INSTALL:
    sklearn (python)
    libblas-dev
    theano (python)
    python-dev

TO USE OUR PICKLED CLASSIFIERS:

from ubc_AI.data import pfdreader
classifier = cPickle.load(open('trained_AI/clfl2_FL.pkl' ,'rb'))
pfdfiles = glob.glob('*.pfd')
#predictions = classifier.predict([pfdreader(f) for f in pfdfiles])
float_number_AI_score=  classifier.report_score([pfdreader(f) for f in pfdfiles])