コード例 #1
0
def test_optimization():
    if os.name != 'nt':
        return
    ema_logging.log_to_stderr(ema_logging.INFO)

    model = FluModel(r'../models', "fluCase")
    ensemble = ModelEnsemble()

    ensemble.model_structure = model
    ensemble.parallel = True

    pop_size = 8
    nr_of_generations = 10
    eps = np.array([1e-3, 1e6])

    stats, pop = ensemble.perform_outcome_optimization(
        obj_function=obj_function_multi,
        algorithm=epsNSGA2,
        reporting_interval=100,
        weights=(MAXIMIZE, MAXIMIZE),
        pop_size=pop_size,
        nr_of_generations=nr_of_generations,
        crossover_rate=0.8,
        mutation_rate=0.05,
        eps=eps)