CausalEGM.util.Sim_Sun_sampler

class CausalEGM.util.Sim_Sun_sampler(batch_size, N=20000, v_dim=200, seed=0)[source]

Sun simulation dataset (continuous treatment) sampler (inherited from Base_sampler).

Parameters:

  • batch_size – Int object denoting the batch size for mini-batch training. Default: 32.

  • N – Sample size. Default: 20000.

  • v_dim – Int object denoting the dimension for covariates. Default: 200.

  • seed – Int object denoting the random seed. Default: 0.

Examples

>>> from CausalEGM import Sim_Sun_sampler
>>> ds = Sim_Sun_sampler(batch_size=32, N=20000, v_dim=200, seed=0)
__init__(batch_size, N=20000, v_dim=200, seed=0)[source]

Methods

__init__(batch_size[, N, v_dim, seed])

create_idx_generator(sample_size[, random_seed])

load_all()

next_batch()