Python API

This page provides documentation for the arginfer Python API.

Running arginfer

arginfer.infer_sim(ts_full, sample_size, iteration=100, thin=20, burn=0, Ne=5000, seq_length=60000.0, mutation_rate=1e-08, recombination_rate=1e-08, outpath='/output', plot=True, verbose=False, verify=False)

Takes msprime tree sequence with record_full_arg= True and - converts tree_sequence to Augmented Tree Sequence (ATS) format. - calculates true likelihood/ prior/ branch length / the number of ancestral and non-ancestral recombinations and returns them at outpath+"/true_values.npy - runs the mcmc and returns samples of ARGs from their posterior.

Parameters

  • ts_full -- msprime tree sequence with record_full_arg= True.

  • sample_size (int) -- The number of sampled genomes.

  • iteration (int) -- The number of MCMC iterations. Must be >20. The default = 100

  • thin (int) -- This specifies how often to write ARG samples to file. By default, the ARG is written every 10 iterations after burn-in.

  • burn (int) -- This specifies how many ARGs to discard as burn-in. Default is 0.

  • Ne (float) -- The effective (diploid) population size for the population. This defaults to 5000 if not specified.

  • seq_length (float) -- The length of the sequences in bases.This defaults to 6e4 if not specified.

  • mutation_rate (float) -- The rate of mutation per base per generation. This defaults to 1e-8 if not specified.

  • recombination_rate (float) -- The rate of recombination per base per generation. This defaults to 1e-8 if not specified.

  • outpath -- The path to store the outputs. This defaults to ./output if not specified.

  • plot (bool) -- plots the trace plots if True

  • verbose (bool) -- verbose.

  • verify (bool) -- for debugging purposes. Checks if the ARG is a valid ATS.

Returns

None. inferred ARGs are stored in outpath.