CausalEGM.util.Sim_Hirano_Imbens_sampler

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

Hirano Imbens 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_Hirano_Imbens_sampler
>>> ds = Sim_Hirano_Imbens_sampler(batch_size=32, N=20000, v_dim=200, seed=0)
__init__(batch_size=32, 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()