esem.emulator.Emulator.batch_stats

Emulator.batch_stats(sample_points, batch_size=1)

Return mean and standard deviation in model predictions over samples, without storing the intermediate predicions in memory to enable evaluating large models over more samples than could fit in memory

Parameters
  • sample_points (pd.DataFrame or array-like) – The parameter values at which to sample the emulator

  • batch_size (int) – The number of samples to calculate in each batch. This can be optimised to fill the available (GPU) memory

Returns

The batch mean and standard deviation with the same type as `self.training_data`