learn cmd: enable specifying --model-path

Author: doronbehar

whereami/__main__.py

@@ -54,6 +54,8 @@ def get_args_parser():
                               help='Change the wifi device to use')
     learn_parser.add_argument('--num_samples', '-n', type=int,
                               default=1, help='Number of samples to take')
+    learn_parser.add_argument('--model_path', '-mp', default=None,
+                                 help='The directory of the model / trained data')
 
     rename = subparsers.add_parser('rename')
 
@@ -77,7 +79,7 @@ def main():
         elif args.command == "predict":
             print(predict(args.input_path, args.model_path, args.device))
         elif args.command == "learn":
-            learn(args.location, args.num_samples, args.device)
+            learn(args.location, args.num_samples, args.device, args.model_path)
         elif args.command == "crossval":
             crossval(path=args.model_path)
         elif args.command in ["locations", "ls"]:

whereami/learn.py

@@ -16,8 +16,8 @@ def write_data(label_path, data):
         f.write("\n")
 
 
-def learn(label, n=1, device=""):
-    path = ensure_whereami_path()
+def learn(label, n=1, device="", model_path=None):
+    path = ensure_whereami_path(model_path)
     label_path = get_label_file(path, label + ".txt")
     for i in tqdm(range(n)):
         if i != 0:

whereami/utils.py

@@ -9,8 +9,8 @@ def get_whereami_path(path=None):
     return os.path.expanduser(path)
 
 
-def ensure_whereami_path():
-    path = get_whereami_path()
+def ensure_whereami_path(path=None):
+    path = get_whereami_path(path)
     if not os.path.exists(path):  # pragma: no cover
         os.makedirs(path)
     return path