earth2mip.lagged_ensembles package#
Submodules#
earth2mip.lagged_ensembles.core module#
- earth2mip.lagged_ensembles.core.done(j, ell, i, lags, n)#
Unused helper function wich says if lag ell and valid_time j are written to in a given iteration i of the loop in lagged_average_simple
This is one way to implement the done criteria which is less easily parallelized. I am leaving it in the code for educational value only.
- earth2mip.lagged_ensembles.core.num(n, ell, j, L)#
- async earth2mip.lagged_ensembles.core.yield_lagged_ensembles(*, observations, forecast, lags=2, n=10)#
Yield centered lagged ensembles
The forecast array has shape (len(observations), n)
The ensemble consist of runs initialized with an offset of (-lags, …, 0, …lags). The ensemble size is therefore
2*lags + =for points within the interior of the array.Supports running in parallel using the
rankandworld_sizeflags