math_problem.py 文件源码

def main(args):
    '''Module main method'''
    random.seed()
    problem = MathProblem()

    database = utils.initialize_database(args, 'MathDatabase')
    database.set_objective_names(['cos', 'sinc'])

    generation = database.properties['highest_population']
    population_size = database.properties['population_size']
    genetic_algorithm = evolution.NSGA(problem, population_size)
    if generation > 0:
        parents, children = utils.load_data(database)
        genetic_algorithm.set_population(parents)
        genetic_algorithm.set_children(children)
    for _ in range(args.iterations):
        generation += 1
        print('Starting generation ' + str(generation))
        genetic_algorithm.iterate()
        database.create_population()
        utils.save_data(genetic_algorithm, database)
        print('=' * (SCREEN_WIDTH - 1))