Ejemplo n.º 1
0
        except KeyError:
            ema_logging.warning("key 'file' not found in policy")
        super(FluModel, self).model_init(policy, kwargs)
        
if __name__ == "__main__":
    ema_logging.log_to_stderr(ema_logging.INFO)
        
    model = FluModel(r'./models/flu', "flucase")
    ensemble = ModelEnsemble()
    ensemble.model_structure = model
    
    #add policies
    policies = [{'name': 'no policy',
                 'file': r'\FLUvensimV1basecase.vpm'},
                {'name': 'static policy',
                 'file': r'\FLUvensimV1static.vpm'},
                {'name': 'adaptive policy',
                 'file': r'\FLUvensimV1dynamic.vpm'}
                ]
    ensemble.policies = policies
    
    #turn on parallel processing
    ensemble.parallel = True 
    
    # run 1000 experiments
    nr_runs = 1000
    results = ensemble.perform_experiments(nr_runs)
    
    # save the results
#     save_results(results, r'./data/{} flu cases.tar.gz'.format(nr_runs))