This repository has been archived on 2021-10-31. You can view files and clone it, but cannot push or open issues or pull requests.
AICup/src/ant_colony.py

56 lines
2.1 KiB
Python
Raw Normal View History

from src.io_tsp import ProblemInstance
import os
# OOT = out-of-time
# Run #0 unknown
# Run #1
# alpha = 0.5, beta = 7, evap = 0.5, weight = instance.best_sol
# if instance.nPoints > 1000: ants = 550 loops = 15
# elif instance.nPoints > 195: ants = 1000 loops = 55
# else: ants = 2500 loops = 150
# pr439=112263 pcb442=53699 d198=16199 fl1577=<OOT> ch130=6332 u1060=247703
# kroA100=21737 eil76=554 rat783=<OOT> lin318=45580
# Run #2
# alpha, beta, evap, weight = (0.9, 8, 0.4, 100_000)
# ants, loops = (900, 6) if instance.nPoints > 1100 \
# else (600, 15) if instance.nPoints > 1000 \
# else (750, 30) if instance.nPoints > 700 \
# else (975, 40) if instance.nPoints > 500 \
# else (1000, 50) if instance.nPoints > 300 \
# else (1100, 70) if instance.nPoints > 195 else (2700, 140)
# pr439=111322 pcb442=52176 d198=16177 ch130=6269 u1060=246085
# kroA100=21665 eil76=550 lin318=43675
# Separate run: fl1577=24238 (132s) rat783=9389 (174s)
dir = "/Users/maggicl/Git/AI2020BsC/c_prob/"
def ant_colony_opt(instance):
fname = dir + instance.name + ".txt"
# write .mat file for C++
if not os.path.exists(fname):
with open(fname, "w") as f:
for i in range(instance.nPoints):
print(" ".join(map(str, instance.dist_matrix[i])), file=f)
alpha, beta, evap, weight = (0.9, 8, 0.4, 100_000)
ants, loops = (1000,7) if instance.nPoints > 1100 \
else (600,15) if instance.nPoints > 1000 \
else (750,30) if instance.nPoints > 700 \
else (975,40) if instance.nPoints > 500 \
else (1000,50) if instance.nPoints > 300 \
else (1100,70) if instance.nPoints > 195 else (2700, 140)
# Call C++ program
cmd = dir + "aco " + str(instance.nPoints) + " " + str(loops) + " " + str(ants) + \
" " + str(alpha) + " " + str(beta) + " " + str(evap) + " " + str(weight) +" < " + fname + " &2>/dev/null"
print(cmd)
solution = eval(os.popen(cmd).read())
solution.append(solution[0])
return solution
if __name__ == "__main__":
ant_colony_opt(ProblemInstance("../problems/eil76.tsp"))