def getLabel(self, value):
feature = FeatureFactory()
feature.createFeature(value, "")
dict = {}
dict['attributes'] = {}
attributes = []
line = feature.datatable[0]
for i in range(len(line)):
dict['attributes'][str(i)] = line[i]
attributes.append(str(i))
res = self.model.predict(dict)
r = max(res.iterkeys(), key=lambda k: res[k])
return r
def getClass(setting, value):
setting = setting.decode("string-escape")
#print setting
classifier = pickle.loads(setting)
feature = FeatureFactory()
feature.createFeature(value, "")
dict = {}
dict['attributes'] = {}
attributes = []
line = feature.datatable[0]
for i in range(len(line)):
dict['attributes'][str(i)] = line[i]
attributes.append(str(i))
res = classifier.predict(dict)
r = max(res.iterkeys(), key=lambda k: res[k])
return r
def __init__(self):
self.path = "./"
self.featureFactory = FeatureFactory()
