run_kcolors.py

import mlrose
import numpy as np

from helpers import algos, model_helper, run_opt


if __name__ == "__main__":
    edges = [(0, 1), (0, 2), (0, 4), (1, 3), (2, 0), (2, 3), (3, 4)]
    fitness = mlrose.MaxKColor(edges)

    sizes = [10, 20, 30, 40, 50, 60]
    for s in sizes:
        problem = mlrose.DiscreteOpt(length = s, fitness_fn = fitness, maximize = True, max_val = 2)
        file_dir = "kcolors/n={}/".format(s)
    
        run_opt.run_rhc(problem, max_attempts=10, max_iters=np.inf, restarts=0, n=10, filedir=file_dir)
    
        schedule = mlrose.GeomDecay(init_temp=1, decay=0.95, min_temp=0.001)
        run_opt.run_sa(problem, decay=0.95, schedule=schedule, max_iters=np.inf, filedir=file_dir)
    
        run_opt.run_ga(problem, m_prob=0.1, pop_size=200, max_iters=np.inf, filedir=file_dir)
    
        run_opt.run_mimic(problem, keep_pct=0.2, pop_size=200, max_iters=np.inf, filedir=file_dir)