Skip to content
Snippets Groups Projects
Commit 9a08da0a authored by Emile Prost's avatar Emile Prost
Browse files

created the sigmoid branch

parent 84a3a586
No related branches found
No related tags found
No related merge requests found
import argparse
import re
import math
import numpy as np
#--------------------------------------------------------------------------
# Cmd line parsing
......@@ -113,11 +114,19 @@ def add_pheromones(tau, paths, costs, Q, n1_size, n2_size, n3_size, n_cbx, n_cby
for i in range(len(costs)):
p = paths[i, :]
cost = costs[i, 0]
tau[0,p[0]] += Q*(-1)*cost
tau[p[0], p[1] + n1_size] += Q*(-1)*cost
tau[p[1] + n1_size, p[2] + n1_size + n2_size] += Q*(-1)*cost
tau[p[2] + n1_size + n2_size, p[3] + n1_size + n2_size + n3_size] += Q*(-1)*cost
tau[p[3] + n1_size + n2_size + n3_size, p[4] + n1_size + n2_size + n3_size + n_cbx] += Q*(-1)*cost
tau[p[4] + n1_size + n2_size + n3_size + n_cbx, p[5] + n1_size + n2_size + n3_size + n_cbx + n_cby] += Q*(-1)*cost
tau[0,p[0]] += Q*(-1)*tools.sigmoid(cost-700)
tau[p[0], p[1] + n1_size] += Q*(-1)*tools.sigmoid(cost-700)
tau[p[1] + n1_size, p[2] + n1_size + n2_size] += Q*(-1)*tools.sigmoid(cost-700)
tau[p[2] + n1_size + n2_size, p[3] + n1_size + n2_size + n3_size] += Q*(-1)*tools.sigmoid(cost-700)
tau[p[3] + n1_size + n2_size + n3_size, p[4] + n1_size + n2_size + n3_size + n_cbx] += Q*(-1)*tools.sigmoid(cost-700)
tau[p[4] + n1_size + n2_size + n3_size + n_cbx, p[5] + n1_size + n2_size + n3_size + n_cbx + n_cby] += Q*(-1)*tools.sigmoid(cost-700)
return tau
#--------------------------------------------------------------------------
# Sigmoid
#--------------------------------------------------------------------------
def sigmoid(x, lmbda=1/30):
res = 1/(1+np.exp(-lmbda*x/100))
return res
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment