Source code for monte_carlo_analysis.strategies.MultipleDistributionsStrategy

"""
**Author** : Robin Camarasa

**Institution** : Erasmus Medical Center

**Position** : PhD student

**Contact** : r.camarasa@erasmusmc.nl

**Date** : 2020-10-16

**Project** : monte_carlo_analysis

**Implement MultipleDistributionsStrategy class**

"""
import numpy as np
from itertools import product
from monte_carlo_analysis.strategies import Strategy
from numba import jit


[docs]class MultipleDistributionsStrategy(Strategy):
[docs]    @staticmethod
    @jit
    def transformation(
            uncertainty_metric_transformation: callable,
            ensemble_output: np.array, counter: list
            ) -> np.array:
        """
        Compute the strategy

        :param ensemble_output: Ensemble error
        """
        # Get the dim size of the output uncertainty map
        uncertainty_map_shape = ensemble_output.shape[2:]

        # Initialize uncertainty metric
        uncertainty_map = np.zeros(uncertainty_map_shape)

        # Loop over the indexes of the output uncertainty metric
        for i in counter:
            uncertainty_map[i] = uncertainty_metric_transformation(
                ensemble_output[(slice(None), slice(None), *i)],
                )
        return uncertainty_map