a b/conf/experiment/treatment_space_sensitivity.yaml 

  1
  
# Configuration file for the treatment space sensitivity experiment





  
  

  2
  
# ==============================================================================





  
  

  3
  





  
  

  4
  
# Defaults





  
  

  5
  
defaults:





  
  

  6
  
  - _self_





  
  

  7
  





  
  

  8
  
  # Simulation settings





  
  

  9
  
  - /simulator: t_simulator #ty_simulator





  
  

  10
  





  
  

  11
  
  # Model parametrization





  
  

  12
  
  - /models@EconML_CausalForestDML: EconML_CausalForestDML





  
  

  13
  
  - /models@EconML_DML: EconML_DML





  
  

  14
  
  - /models@EconML_DMLOrthoForest: EconML_DMLOrthoForest





  
  

  15
  
  - /models@EconML_DRLearner: EconML_DRLearner





  
  

  16
  
  - /models@EconML_DROrthoForest: EconML_DROrthoForest





  
  

  17
  
  - /models@EconML_ForestDRLearner: EconML_ForestDRLearner





  
  

  18
  
  - /models@EconML_LinearDML: EconML_LinearDML





  
  

  19
  
  - /models@EconML_LinearDRLearner: EconML_LinearDRLearner





  
  

  20
  
  - /models@EconML_SparseLinearDML: EconML_SparseLinearDML





  
  

  21
  
  - /models@EconML_SparseLinearDRLearner: EconML_SparseLinearDRLearner #EconML_SparseLinearDRLearner





  
  

  22
  
  - /models@EconML_SLearner_Lasso: EconML_SLearner_Lasso





  
  

  23
  
  - /models@EconML_TLearner_Lasso: EconML_TLearner_Lasso





  
  

  24
  
  - /models@EconML_XLearner_Lasso: EconML_XLearner_Lasso





  
  

  25
  





  
  

  26
  
  - /models@Torch_SLearner: Torch_SLearner





  
  

  27
  
  - /models@Torch_TLearner: Torch_TLearner





  
  

  28
  
  - /models@Torch_XLearner: Torch_XLearner





  
  

  29
  
  - /models@Torch_DRLearner: Torch_DRLearner





  
  

  30
  
  - /models@Torch_RLearner: Torch_RLearner





  
  

  31
  
  - /models@Torch_TARNet: Torch_TARNet





  
  

  32
  
  - /models@Torch_DragonNet: Torch_DragonNet





  
  

  33
  
  - /models@Torch_ULearner: Torch_ULearner





  
  

  34
  
  - /models@Torch_RALearner: Torch_RALearner





  
  

  35
  
  - /models@Torch_PWLearner: Torch_PWLearner





  
  

  36
  
  - /models@Torch_FlexTENet: Torch_FlexTENet





  
  

  37
  
  - /models@Torch_CRFNet_0_01: Torch_CRFNet_0_01





  
  

  38
  
  - /models@Torch_CRFNet_0_001: Torch_CRFNet_0_001





  
  

  39
  
  - /models@Torch_CRFNet_0_0001: Torch_CRFNet_0_0001





  
  

  40
  
  - /models@Torch_ActionNet: Torch_ActionNet





  
  

  41
  
  





  
  

  42
  
  - /models@DiffPOLearner: DiffPOLearner





  
  

  43
  





  
  

  44
  





  
  

  45
  





  
  

  46
  
# EXPERIMENT AND DATA





  
  

  47
  
# ==============================================================================





  
  

  48
  
experiment_name: "treatment_space_sensitivity"





  
  

  49
  
# ==============================================================================





  
  

  50
  





  
  

  51
  
# EXPERIMENTAL KNOB





  
  

  52
  
# ==============================================================================





  
  

  53
  
# Cohort sizes to be tested





  
  

  54
  
num_Ts: [2,3,5,10]





  
  

  55
  
# ==============================================================================





  
  

  56
  





  
  

  57