def make_plan(designs, X_source_dist, X_target_dist):
plan = Plan()
for d in designs:
name = d[0]
dgn = d[1]
estr = d[2]
design_kwargs = d[3]
design_kwargs['source'] = X_source_dist
design_kwargs['target'] = X_target_dist
plan.add_design(name, dgn, estr, design_kwargs)
plan.add_evaluator('ATEError', evalr.ATEError)
return plan
def make_plan(designs):
plan = Plan()
for name, dgn, estr in designs:
plan.add_design(name, dgn, estr)
plan.add_evaluator('ATEError', evalr.ATEError)
plan.add_evaluator('ITEBias', evalr.ITEBias)
plan.add_evaluator('ITEMSE', evalr.ITEMSE)
plan.add_evaluator('CovariateMSE', evalr.CovariateMSE)
plan.add_evaluator('ATECovers', evalr.ATECovers)
plan.add_evaluator('CISize', evalr.CISize)
return plan