mushi.utils.fold

fold(x)

Transform SFS to folded SFS.

Parameters:

func – loss function name from loss_functions module

Return type:

Array

Examples

>>> from mushi import utils
>>> import jax.numpy as np

>>> sfs = np.array([1000, 100, 10])

>>> utils.fold(sfs)
Array([1010,  100], dtype=int64)