From aac480c794695627a5a641263c839a5938e3d329 Mon Sep 17 00:00:00 2001 From: mattianeroni <54416801+mattianeroni@users.noreply.github.com> Date: Thu, 6 May 2021 19:09:59 +0200 Subject: [PATCH] Update utils.py --- tspy/solvers/utils.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tspy/solvers/utils.py b/tspy/solvers/utils.py index 0a1a00c..5371c9c 100644 --- a/tspy/solvers/utils.py +++ b/tspy/solvers/utils.py @@ -1,5 +1,13 @@ +import numpy as np + def get_cost(tour,tsp): ans = 0 for i in range(len(tour)-1): ans += tsp.mat[tour[i],tour[i+1]] return ans + + + +def fast_cost (tour, tsp): + t = np.asarray(tour) + return tsp.mat[t, np.roll(tour, -1)].sum()