def prob_dist(self, x): logged_ps = naive_bayes.calculate(self.nbayes_model, self._map_x(x)) return dict(zip(logged_ps.keys(), [math.exp(x_i) for x_i in logged_ps.values()]))
def prob_dist(self, x): return naive_bayes.calculate(self.nbayes_model, self._map_x(x))