Index

_

_abc_impl (BaseCATENet attribute) (DragonNet attribute), [1] (DRNet attribute), [1] (FlexTENet attribute), [1] (OffsetNet attribute), [1] (PseudoOutcomeNet attribute), [1] (PWNet attribute), [1] (RANet attribute), [1] (RNet attribute), [1] (SNet attribute), [1] (SNet1 attribute), [1] (SNet2 attribute), [1] (SNet3 attribute), [1] (TARNet attribute), [1] (TNet attribute), [1] (XNet attribute), [1] _backward_hooks (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _buffers (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _check_inputs() (BaseCATENet static method) _check_tensor() (BaseCATEEstimator method) (BasicNet method) (PropensityNet method) _compute_ortho_penalty_asymmetric() (in module catenets.models.jax.flextenet) _compute_penalty() (in module catenets.models.jax.flextenet) _compute_penalty_l2() (in module catenets.models.jax.flextenet) _concatenate_representations() (in module catenets.models.jax.disentangled_nets) _create_extended_matrices() (SLearner method), [1] _first_step() (DRLearner method), [1] (PseudoOutcomeLearner method) (PWLearner method), [1] (RALearner method), [1] (RLearner method), [1] (ULearner method), [1] (XLearner method), [1] _forward() (BasicDragonNet method) (SNet method), [1] _forward_hooks (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _forward_pre_hooks (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _generate_po_estimator() (PseudoOutcomeLearner method) _generate_propensity_estimator() (PseudoOutcomeLearner method) _generate_te_estimator() (PseudoOutcomeLearner method) _get_absolute_rowsums() (in module catenets.models.jax.disentangled_nets) _get_cos_reg() (in module catenets.models.jax.flextenet) _get_first_stage_pos() (in module catenets.models.jax.xnet) _get_predict_function() (BaseCATENet method) (FlexTENet method), [1] (OffsetNet method), [1] (PseudoOutcomeNet method), [1] (RNet method), [1] (SNet method), [1] (SNet1 method), [1] (SNet2 method), [1] (SNet3 method), [1] (TNet method), [1] (XNet method), [1] _get_train_function() (BaseCATENet method) (FlexTENet method), [1] (OffsetNet method), [1] (PseudoOutcomeNet method), [1] (RNet method), [1] (SNet method), [1] (SNet1 method), [1] (SNet2 method), [1] (SNet3 method), [1] (TNet method), [1] (XNet method), [1] _get_transformation_function() (in module catenets.models.jax.transformation_utils) _impute_pos() (PseudoOutcomeLearner method) _impute_propensity() (PseudoOutcomeLearner method) _impute_unconditional_mean() (PseudoOutcomeLearner method) _is_full_backward_hook (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1]

_load_state_dict_post_hooks (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _load_state_dict_pre_hooks (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _maximum_mean_discrepancy() (BasicDragonNet method) (SNet method), [1] _modules (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _non_persistent_buffers_set (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _ortho_penalty_asymmetric() (FlexTENet method), [1] _ortho_reg() (SNet method), [1] _parameters (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _plug_in (TLearner attribute) _second_step() (DRLearner method), [1] (PseudoOutcomeLearner method) (PWLearner method), [1] (RALearner method), [1] (RLearner method), [1] (ULearner method), [1] (XLearner method), [1] _state_dict_hooks (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1] _step() (BasicDragonNet method) (DragonNet method), [1] (SNet method), [1] (TARNet method), [1] _train_and_predict_first_stage() (in module catenets.models.jax.pseudo_outcome_nets) _train_and_predict_r_stage1() (in module catenets.models.jax.rnet) _train_tnet_jointly() (in module catenets.models.jax.tnet)

A

abs_error_ATE() (in module catenets.experiment_utils.torch_metrics)

add() (in module catenets.logger) aipw_te_transformation() (in module catenets.models.jax.transformation_utils)

B

BaseCATEEstimator (class in catenets.models.torch.base) BaseCATENet (class in catenets.models.jax.base)

BasicDragonNet (class in catenets.models.torch.representation_nets) BasicNet (class in catenets.models.torch.base) benchmark() (in module catenets.models.torch.utils.decorators)

C

catenets module catenets.datasets module catenets.datasets.dataset_acic2016 module catenets.datasets.dataset_ihdp module catenets.datasets.dataset_twins module catenets.datasets.network module catenets.experiment_utils module catenets.experiment_utils.base module catenets.experiment_utils.simulation_utils module catenets.experiment_utils.tester module catenets.experiment_utils.torch_metrics module catenets.logger module catenets.models module catenets.models.constants module catenets.models.jax module catenets.models.jax.base module catenets.models.jax.disentangled_nets module catenets.models.jax.flextenet module catenets.models.jax.model_utils module catenets.models.jax.offsetnet module catenets.models.jax.pseudo_outcome_nets module catenets.models.jax.representation_nets module catenets.models.jax.rnet module

catenets.models.jax.snet module catenets.models.jax.tnet module catenets.models.jax.transformation_utils module catenets.models.jax.xnet module catenets.models.torch module catenets.models.torch.base module catenets.models.torch.flextenet module catenets.models.torch.pseudo_outcome_nets module catenets.models.torch.representation_nets module catenets.models.torch.slearner module catenets.models.torch.snet module catenets.models.torch.tlearner module catenets.models.torch.utils module catenets.models.torch.utils.decorators module catenets.models.torch.utils.model_utils module catenets.models.torch.utils.transformations module catenets.models.torch.utils.weight_utils module catenets.version module check_input_train() (in module catenets.models.torch.utils.decorators) check_shape_1d_data() (in module catenets.models.jax.model_utils) check_X_is_np() (in module catenets.models.jax.model_utils) compute_importance_weights() (in module catenets.models.torch.utils.weight_utils) compute_ipw() (in module catenets.models.torch.utils.weight_utils) compute_matching_weights() (in module catenets.models.torch.utils.weight_utils) compute_overlap_weights() (in module catenets.models.torch.utils.weight_utils) compute_trunc_ipw() (in module catenets.models.torch.utils.weight_utils) create_log_and_print_function() (in module catenets.logger) critical() (in module catenets.logger)

D

debug() (in module catenets.logger) DenseW() (in module catenets.models.jax.flextenet) download_gdrive_if_needed() (in module catenets.datasets.network) download_http_if_needed() (in module catenets.datasets.network) download_if_needed() (in module catenets.datasets.network) dr_transformation_cate() (in module catenets.models.torch.utils.transformations) DragonNet (class in catenets.models.jax) (class in catenets.models.jax.representation_nets) (class in catenets.models.torch) (class in catenets.models.torch.representation_nets)

DRLearner (class in catenets.models.torch) (class in catenets.models.torch.pseudo_outcome_nets) DRNet (class in catenets.models.jax) (class in catenets.models.jax.pseudo_outcome_nets)

E

elementwise_parallel() (in module catenets.models.jax.flextenet) elementwise_split() (in module catenets.models.jax.flextenet) ElementWiseParallelActivation (class in catenets.models.torch.flextenet) ElementWiseSplitActivation (class in catenets.models.torch.flextenet) error() (in module catenets.logger)

eval_abs_error_ate() (in module catenets.experiment_utils.base) eval_mse() (in module catenets.experiment_utils.base) eval_mse_model() (in module catenets.experiment_utils.base) eval_root_mse() (in module catenets.experiment_utils.base) evaluate_treatments_model() (in module catenets.experiment_utils.tester)

F

fit() (BaseCATEEstimator method) (BaseCATENet method) (BasicDragonNet method) (BasicNet method) (FlexTENet method), [1] (PropensityNet method) (PseudoOutcomeLearner method) (PseudoOutcomeNet method), [1] (RNet method), [1] (SLearner method), [1] (SNet method), [1] (TLearner method), [1] fit_and_select_params() (BaseCATENet method) FlexTELinearLayer (class in catenets.models.torch.flextenet) FlexTENet (class in catenets.models.jax) (class in catenets.models.jax.flextenet) (class in catenets.models.torch) (class in catenets.models.torch.flextenet)

FlexTENetArchitecture() (in module catenets.models.jax.flextenet) FlexTEOutputLayer (class in catenets.models.torch.flextenet) FlexTESplitLayer (class in catenets.models.torch.flextenet) forward() (BaseCATEEstimator method) (BasicNet method) (ElementWiseParallelActivation method) (ElementWiseSplitActivation method) (FlexTELinearLayer method) (FlexTEOutputLayer method) (FlexTESplitLayer method) (PropensityNet method) (RepresentationNet method)

G

generate_score() (in module catenets.experiment_utils.tester) get_acic_covariates() (in module catenets.datasets.dataset_acic2016) get_acic_orig_filenames() (in module catenets.datasets.dataset_acic2016) get_acic_orig_outcomes() (in module catenets.datasets.dataset_acic2016) get_all_pseudoout_models() (in module catenets.experiment_utils.base) get_all_snets() (in module catenets.experiment_utils.base)

get_all_twostep_models() (in module catenets.experiment_utils.base) get_importance_weights() (PropensityNet method) get_model_set() (in module catenets.experiment_utils.base) get_multivariate_normal_params() (in module catenets.experiment_utils.simulation_utils) get_one_data_set() (in module catenets.datasets.dataset_ihdp) get_set_normal_covariates() (in module catenets.experiment_utils.simulation_utils)

H

heads_l2_penalty() (in module catenets.models.jax.model_utils)

ht_te_transformation() (in module catenets.models.jax.transformation_utils)

I

info() (in module catenets.logger)

L

load() (in module catenets.datasets) (in module catenets.datasets.dataset_acic2016) (in module catenets.datasets.dataset_ihdp) (in module catenets.datasets.dataset_twins) load_data_npz() (in module catenets.datasets.dataset_ihdp)

load_raw() (in module catenets.datasets.dataset_ihdp) loss() (BasicDragonNet method) (FlexTENet method), [1] (PropensityNet method) (SNet method), [1]

M

make_val_split() (in module catenets.models.jax.model_utils) (in module catenets.models.torch.utils.model_utils) mmd2_lin() (in module catenets.models.jax.representation_nets) module catenets catenets.datasets catenets.datasets.dataset_acic2016 catenets.datasets.dataset_ihdp catenets.datasets.dataset_twins catenets.datasets.network catenets.experiment_utils catenets.experiment_utils.base catenets.experiment_utils.simulation_utils catenets.experiment_utils.tester catenets.experiment_utils.torch_metrics catenets.logger catenets.models catenets.models.constants catenets.models.jax catenets.models.jax.base catenets.models.jax.disentangled_nets catenets.models.jax.flextenet catenets.models.jax.model_utils catenets.models.jax.offsetnet catenets.models.jax.pseudo_outcome_nets catenets.models.jax.representation_nets catenets.models.jax.rnet catenets.models.jax.snet catenets.models.jax.tnet catenets.models.jax.transformation_utils catenets.models.jax.xnet catenets.models.torch catenets.models.torch.base catenets.models.torch.flextenet catenets.models.torch.pseudo_outcome_nets catenets.models.torch.representation_nets catenets.models.torch.slearner catenets.models.torch.snet catenets.models.torch.tlearner catenets.models.torch.utils catenets.models.torch.utils.decorators catenets.models.torch.utils.model_utils catenets.models.torch.utils.transformations catenets.models.torch.utils.weight_utils catenets.version

mu0_AISTATS() (in module catenets.experiment_utils.simulation_utils) mu0_hg() (in module catenets.experiment_utils.simulation_utils) mu1_additive() (in module catenets.experiment_utils.simulation_utils) mu1_AISTATS() (in module catenets.experiment_utils.simulation_utils) mu1_hg() (in module catenets.experiment_utils.simulation_utils)

N

normal_covariate_model() (in module catenets.experiment_utils.simulation_utils)

O

OffsetNet (class in catenets.models.jax) (class in catenets.models.jax.offsetnet)

OutputHead() (in module catenets.models.jax.base)

P

predict() (BaseCATEEstimator method) (BaseCATENet method) (BasicDragonNet method) (FlexTENet method), [1] (PseudoOutcomeLearner method) (PseudoOutcomeNet method), [1] (RNet method), [1] (SLearner method), [1] (SNet method), [1] (TLearner method), [1] (XLearner method), [1] (XNet method), [1] predict_flextenet() (in module catenets.models.jax.flextenet) predict_offsetnet() (in module catenets.models.jax.offsetnet) predict_snet() (in module catenets.models.jax.snet) predict_snet1() (in module catenets.models.jax.representation_nets) predict_snet2() (in module catenets.models.jax.representation_nets) predict_snet3() (in module catenets.models.jax.disentangled_nets) predict_snet_noprop() (in module catenets.models.jax.snet) predict_t_net() (in module catenets.models.jax.tnet)

predict_wrapper() (in module catenets.models.torch.utils.model_utils) predict_x_net() (in module catenets.models.jax.xnet) prepare_ihdp_data() (in module catenets.datasets.dataset_ihdp) preprocess() (in module catenets.datasets.dataset_acic2016) (in module catenets.datasets.dataset_twins) preprocess_acic_orig() (in module catenets.datasets.dataset_acic2016) preprocess_simu() (in module catenets.datasets.dataset_acic2016) print_score() (in module catenets.experiment_utils.tester) propensity_AISTATS() (in module catenets.experiment_utils.simulation_utils) propensity_constant() (in module catenets.experiment_utils.simulation_utils) propensity_hg() (in module catenets.experiment_utils.simulation_utils) PropensityNet (class in catenets.models.torch.base) PseudoOutcomeLearner (class in catenets.models.torch.pseudo_outcome_nets) PseudoOutcomeNet (class in catenets.models.jax) (class in catenets.models.jax.pseudo_outcome_nets) pw_transformation_cate() (in module catenets.models.torch.utils.transformations) PWLearner (class in catenets.models.torch) (class in catenets.models.torch.pseudo_outcome_nets) PWNet (class in catenets.models.jax) (class in catenets.models.jax.pseudo_outcome_nets)

R

ra_te_transformation() (in module catenets.models.jax.transformation_utils) ra_transformation_cate() (in module catenets.models.torch.utils.transformations) RALearner (class in catenets.models.torch) (class in catenets.models.torch.pseudo_outcome_nets) RANet (class in catenets.models.jax) (class in catenets.models.jax.pseudo_outcome_nets)

remove() (in module catenets.logger) ReprBlock() (in module catenets.models.jax.base) RepresentationNet (class in catenets.models.torch.base) RLearner (class in catenets.models.torch) (class in catenets.models.torch.pseudo_outcome_nets) RNet (class in catenets.models.jax) (class in catenets.models.jax.rnet)

S

score() (BaseCATEEstimator method) (BaseCATENet method) simulate_treatment_setup() (in module catenets.experiment_utils.simulation_utils) SLearner (class in catenets.models.torch) (class in catenets.models.torch.slearner) SNet (class in catenets.models.jax) (class in catenets.models.jax.snet) (class in catenets.models.torch) (class in catenets.models.torch.snet)

SNet1 (class in catenets.models.jax) (class in catenets.models.jax.representation_nets) SNet2 (class in catenets.models.jax) (class in catenets.models.jax.representation_nets) SNet3 (class in catenets.models.jax) (class in catenets.models.jax.disentangled_nets) SplitLayerAsymmetric() (in module catenets.models.jax.flextenet) sqrt_PEHE() (in module catenets.experiment_utils.torch_metrics)

T

TARNet (class in catenets.models.jax) (class in catenets.models.jax.representation_nets) (class in catenets.models.torch) (class in catenets.models.torch.representation_nets) TEOutputLayerAsymmetric() (in module catenets.models.jax.flextenet) TLearner (class in catenets.models.torch) (class in catenets.models.torch.tlearner) TNet (class in catenets.models.jax) (class in catenets.models.jax.tnet) trace() (in module catenets.logger) traceback() (in module catenets.logger) traceback_and_raise() (in module catenets.logger) train_flextenet() (in module catenets.models.jax.flextenet) train_offsetnet() (in module catenets.models.jax.offsetnet) train_output_net_only() (in module catenets.models.jax.base) train_pseudooutcome_net() (in module catenets.models.jax.pseudo_outcome_nets) train_r_net() (in module catenets.models.jax.rnet) train_r_stage2() (in module catenets.models.jax.rnet) train_snet() (in module catenets.models.jax.snet) train_snet1() (in module catenets.models.jax.representation_nets) train_snet2() (in module catenets.models.jax.representation_nets) train_snet3() (in module catenets.models.jax.disentangled_nets) train_snet_noprop() (in module catenets.models.jax.snet) train_tnet() (in module catenets.models.jax.tnet)

train_wrapper() (in module catenets.models.torch.utils.model_utils) train_x_net() (in module catenets.models.jax.xnet) training (BaseCATEEstimator attribute) (BasicDragonNet attribute) (BasicNet attribute) (DragonNet attribute), [1] (DRLearner attribute), [1] (ElementWiseParallelActivation attribute) (ElementWiseSplitActivation attribute) (FlexTELinearLayer attribute) (FlexTENet attribute), [1] (FlexTEOutputLayer attribute) (FlexTESplitLayer attribute) (PropensityNet attribute) (PseudoOutcomeLearner attribute) (PWLearner attribute), [1] (RALearner attribute), [1] (RepresentationNet attribute) (RLearner attribute), [1] (SLearner attribute), [1] (SNet attribute), [1] (TARNet attribute), [1] (TLearner attribute), [1] (ULearner attribute), [1] (XLearner attribute), [1]

U

u_transformation_cate() (in module catenets.models.torch.utils.transformations) ULearner (class in catenets.models.torch) (class in catenets.models.torch.pseudo_outcome_nets)

unarchive_if_needed() (in module catenets.datasets.network) uniform_covariate_model() (in module catenets.experiment_utils.simulation_utils)

W

warning() (in module catenets.logger)

X

XLearner (class in catenets.models.torch) (class in catenets.models.torch.pseudo_outcome_nets)

XNet (class in catenets.models.jax) (class in catenets.models.jax.xnet)