def get_pre_Q(cls, log_counts, codon_distn, ts, tv, syn, nonsyn, compo, asym_compo, natural_theta):
cls.check_theta(natural_theta)
kimura_d = natural_theta[0]
kappa = natural_theta[1]
omega = natural_theta[2]
nt_distn = markovutil.ratios_to_distn(natural_theta[3:6])
pre_Q = fmutsel.get_pre_Q_unconstrained(
log_counts, ts, tv, syn, nonsyn, compo, asym_compo, kimura_d, nt_distn, kappa, omega
)
return pre_Q
def get_mixture(cls,
log_counts, codon_distn,
ts, tv, syn, nonsyn, compo, asym_compo,
natural_theta,
):
cls.check_theta(natural_theta)
p0 = natural_theta[0]
p1 = 1 - p0
first_omega = natural_theta[1]
second_omega = natural_theta[2]
kimura_d = natural_theta[3]
kappa = natural_theta[4]
nt_distn = markovutil.ratios_to_distn(natural_theta[5:5+3])
first_pre_Q = fmutsel.get_pre_Q_unconstrained(
log_counts,
ts, tv, syn, nonsyn, compo, asym_compo,
kimura_d, nt_distn, kappa, first_omega,
)
second_pre_Q = fmutsel.get_pre_Q_unconstrained(
log_counts,
ts, tv, syn, nonsyn, compo, asym_compo,
kimura_d, nt_distn, kappa, second_omega,
)
return (p0, p1), (first_pre_Q, second_pre_Q)
