Merge pull request #958 from tue-robotics/MatthijsBurgh-patch-1

(find_my_mates) fix beun code van @LarsJanssenTUe
tue-robotics · Dec 14, 2019 · a15d393 · a15d393
2 parents ef94381 + 7841e53
commit a15d393
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/challenge_find_my_mates/src/challenge_find_my_mates/cluster.py b/challenge_find_my_mates/src/challenge_find_my_mates/cluster.py
@@ -5,6 +5,8 @@
 
 import matplotlib.pyplot as plt
 
+from os.path import expanduser
+
 from sklearn.cluster import KMeans
 
 
@@ -69,5 +71,5 @@ def cluster_people(people_dicts, room_center, plot=False):
     locations = zip(xs2, ys2)
     pprint.pprint(locations)
 
-    with open('/home/kmeans_output.pickle', 'w') as dumpfile:
+    with open(expanduser('~') + '/kmeans_output.pickle', 'w') as dumpfile:
         pickle.dump(clustered_ppl, dumpfile)