optimiser

anirudhpillai · Jan 27, 2018 · d1d8e8c · d1d8e8c
1 parent a372d62
commit d1d8e8c
Show file tree

Hide file tree

Showing 4 changed files with 1,018 additions and 999 deletions.
diff --git a/answer.py b/answer.py
@@ -3,6 +3,7 @@
 import pandas as pd
 from sklearn.cluster import KMeans
 import numpy as np
+import sys
 
 
 MAX_DELIVERIES = 10
@@ -11,7 +12,7 @@
 PRICE_PER_KM = 0.061
 MAX_TIME_PER_TRUCK = 10
 
-TOTAL_CLUSTERS = 40
+TOTAL_CLUSTERS = 51
 
 
 def distance(x1, y1, x2, y2):
@@ -145,8 +146,6 @@ def answer():
     kmeans = KMeans(n_clusters=TOTAL_CLUSTERS, random_state=0)
     kmeans.fit(orders_to_cluster)
 
-    print(len(orders))
-
     # ans_list = bruteforce(orders)
     ans_list = clustered(orders, kmeans)
 
@@ -159,4 +158,8 @@ def answer():
             writer.writerow(row)
 
 
+if len(sys.argv) > 1:
+    TOTAL_CLUSTERS = int(sys.argv[1])
+
+
 answer()