naivbayes.py
#bayes classifer, feeds in dictionary objects.
from decimal import *
getcontext().prec = 5
class NBC (object):
def __init__(self, *keys):
self.keys = keys
self.data = {elem:[] for elem in self.keys}
def probofkey(self, key, outcome):
return Decimal(len([elem for elem in self.data[key] if elem == outcome]))/Decimal(len(self.data[key]))