Beispiel #1
0
# For quick test simulations, this is set to a very low value
static_params = {'Base_Population_Scale_Factor': 0.05}
cfg.set_params(static_params)
cpn.set_params(static_params)

# Prepare templates
templates = TemplateHelper()
table_base = {
    'ACTIVE_TEMPLATES': [cfg, cpn, demog_pfa, demog_acc, demog_asrt],
    'TAGS': {
        'Scenario': 'StatusQuo_Baseline',
        'pyOptimTool': None
    }
}

config_builder = DTKConfigBuilder()
config_builder.ignore_missing = True


def constrain_sample(sample):
    if 'Pr Ex Trns Male LOW' and 'Pr Ex Trns Male MED' in sample:
        sample['Pr Ex Trns Male LOW'] = min(
            [sample['Pr Ex Trns Male LOW'], sample['Pr Ex Trns Male MED']])
    if 'Pr Ex Trns Fem LOW' and 'Pr Ex Trns Fem MED' in sample:
        sample['Pr Ex Trns Fem LOW'] = min(
            [sample['Pr Ex Trns Fem LOW'], sample['Pr Ex Trns Fem MED']])

    return sample


def map_sample_to_model_input(config_builder, s):
from dtk.utils.core.DTKConfigBuilder import DTKConfigBuilder
from dtk.utils.builders.regression import RegressionSuiteBuilder

test_names = ['27_Vector_Sandbox', '10_Vector_Namawala_Oviposition']

cb = DTKConfigBuilder()
builder = RegressionSuiteBuilder(test_names)

run_sim_args = {
    'config_builder': cb,
    'exp_builder': builder,
    'exp_name': 'Regression Suite'
}