Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
Download

Repository for a workshop on Bayesian statistics

1430 views
Tweet
Share
Share
1
"""This file contains code for use with "Think Bayes",

2
by Allen B. Downey, available from greenteapress.com

3


4
Copyright 2012 Allen B. Downey

5
License: GNU GPLv3 http://www.gnu.org/licenses/gpl.html

6
"""

7


8
from __future__ import print_function, division

9


10
import thinkbayes

11
import thinkplot

12


13


14
class Train(thinkbayes.Suite):

15
    """Represents hypotheses about how many trains the company has.

16


17
    The likelihood function for the train problem is the same as

18
    for the Dice problem.

19
    """

20
    def Likelihood(self, data, hypo):

21
        """Computes the likelihood of the data under the hypothesis.

22


23
        hypo: number of trains the carrier operates

24
        data: the number of the observed train

25
        """

26
        # fill this in!

27
        return 1

28


29


30


31
def main():

32
    hypos = range(100, 1001)

33
    suite = Train(hypos)

34


35
    suite.Update(321)

36


37
    thinkplot.PrePlot(1)

38
    thinkplot.Pmf(suite)

39
    thinkplot.Show(xlabel='Number of trains',

40
                   ylabel='Probability',

41
                   legend=False)

42


43


44
if __name__ == '__main__':

45
    main()

46


47