jweinst1
1/13/2016 - 9:04 AM

naivbayes.py

naivbayes.py

#bayes classifer, feeds in dictionary objects.

class NBC (object):
	
	def __init__(self, *keys):
		self.keys = keys
		self.data = {elem:[] for elem in self.keys}
	def probofkey(self, key, outcome):
		return len([elem for elem in self.data[key] if elem == outcome])/len(self.data[key])