def main(): all_births, first_births, later_births = descriptive.MakeTables() later_births.pmf.name = 'other babies' print " All births: early = {:.1%}".format(ProbEarly(all_births.pmf)) print " on-time = {:.1%}".format(ProbOnTime(all_births.pmf)) print " late = {:.1%}".format(ProbLate(all_births.pmf)) PrintRelativeRisks(first_births.pmf, later_births.pmf)
def testVar(self): pool, firsts, others = descriptive.MakeTables() descriptive.Process(pool, 'Live births') descriptive.Process(firsts, 'First babies') descriptive.Process(others, 'Others') self.assertAlmostEquals(firsts.var, 7.79294720207) self.assertAlmostEquals(others.var, 6.84123839078) self.assertAlmostEquals(pool.mu, 38.5605596852) self.assertAlmostEquals(pool.var, 7.3018637882)
def main(): """ <39 """ pool, firsts, others = descriptive.MakeTables("res") pmf = pool.pmf print("before pro(39) = ", pmf.Prob(39)) for v in range(0, 39): try: pmf.Remove(v) except Exception as e: continue pmf.Normalize() print("after pro(39) = ", pmf.Prob(39))
import descriptive import Pmf def filtered_pmf(filter_function, pmf): filtered_items = filter(filter_function, pmf.Items()) filtered_pmf = Pmf.MakePmfFromDict(dict(filtered_items)) filtered_pmf.Normalize() return filtered_pmf def prob_birth_in_next_week(week, pmf): new_pmf = filtered_pmf(lambda x: x[0] > week, pmf) return new_pmf.Prob(week + 1) pool, firsts, others = descriptive.MakeTables() later_first_births = prob_birth_in_next_week(38, firsts.pmf) later_other_births = prob_birth_in_next_week(38, others.pmf) print 'chance of first birth in week 39 %f' % later_first_births print 'chance of other birth in week 39 %f' % later_other_births
def main(): pool, firsts, others = descriptive.MakeTables() ComputeRelativeRisk(firsts.pmf, others.pmf)
def main(): # get the data pool, firsts, others = descriptive.MakeTables() Test(pool, firsts, others, num_trials=1000)
def main(): pool, firsts, others = descriptive.MakeTables(data_dir='./dataset') RelativeRisk(firsts, others) MakeFigure(firsts, others)
def main(): pool, firsts, others = descriptive.MakeTables(data_dir='./dataset') ComputeRelativeRisk(firsts.pmf, others.pmf)
def main(): pool, firsts, others = descriptive.MakeTables() RelativeRisk(firsts, others) MakeFigure(firsts, others)
def main(): pool, firsts, others = descriptive.MakeTables() first_cond, other_cond = RelativeRisk(firsts, others) MakeFigurePmf(first_cond, other_cond)
def main(): pool, firsts, others = descriptive.MakeTables() firsts.pmf.name = "first" print "Probabilty at 39 weeks for first baby: {:.1%}".format(CondPmf(firsts.pmf,39).Prob(39)) print "Probabilty at 39 weeks for later babies: {:.1%}".format(CondPmf(others.pmf,39).Prob(39)) ShowPlot(firsts, others)