File: seqsim_tree_sim.rst

package info (click to toggle)
python-cogent 1.5.3-2
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 16,424 kB
  • ctags: 24,343
  • sloc: python: 134,200; makefile: 100; ansic: 17; sh: 10
file content (43 lines) | stat: -rw-r--r-- 1,347 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
Seqsim Simple Tree Simulation
=============================

.. sectionauthor:: Julia Goodrich

This is an example of how to use the birth-death model in cogent to simulate
a tree.

.. doctest::

    >>> from cogent.seqsim.birth_death import BirthDeathModel, ExtinctionError,\
    ... TooManyTaxaError

Create a model with specific death probabilities per timestep using 
``BirthProb`` and ``DeathProb``, and ``TimePerStep``. The desired maximum 
number of taxa on the tree can be set using ``MaxTaxa``.


.. doctest::

    >>> b = BirthDeathModel(BirthProb=0.2, DeathProb=0.1, TimePerStep=0.03, 
    ... MaxTaxa=20)

To simulate a tree with an exact number of taxa, use the following loop. 
The ``exact`` flag raises a ``TooManyTaxaError`` exception if the call produces
the wrong number of taxa (e.g. because too many things died in the same 
timestep). An ``ExtinctionError`` is raised if the ancestral node dies off, or 
all the nodes die off.


.. doctest::

    >>> while True:
    ...     try:
    ...         t = b(exact=True)
    ...     except (ExtinctionError, TooManyTaxaError):
    ...         pass
    ...     else:
    ...         break

t now contains a ``RangeNode`` tree object that was returned, and can be used 
the same ways any other tree object in cogent is used. For instance, ``seqsim`` 
can be used to evolve alignments.