refine arguments

2 files changed, 5 insertions, 5 deletions

diff --git a/reduce.py b/reduce.py
index 2751124..6c12519 100755
--- a/reduce.py
+++ b/reduce.py
@@ -48,7 +48,7 @@ def iterateSubDirectory(oldroot, newroot, level):
 if __name__ == '__main__':
     parser = ArgumentParser(description='Reduce the levels of categories.')
     parser.add_argument('--level', action='store', default=2, \
-                            help='reduce to n levels of index')
+                            help='reduce to n levels of index', type=int)
     parser.add_argument('origdir', action='store', \
                             help='original index directory')
     parser.add_argument('--destdir', action='store', \
@@ -57,5 +57,5 @@ if __name__ == '__main__':
 
     args = parser.parse_args()
     print(args)
-    iterateSubDirectory(args.origdir, args.destdir, int(args.level))
+    iterateSubDirectory(args.origdir, args.destdir, args.level)
     print('done')
diff --git a/tryprune.py b/tryprune.py
index 7d0a32d..ee0acbd 100644
--- a/tryprune.py
+++ b/tryprune.py
@@ -83,15 +83,15 @@ if __name__ == '__main__':
 
     parser.add_argument('--mergenumber', action='store', \
                             help='number of documents to be merged', \
-                            default='10')
+                            default=10, type=int)
 
     parser.add_argument('-k', action='store', \
                             help='k parameter of k mixture model prune', \
-                            default='3')
+                            default=3, type=int)
 
     parser.add_argument('--CDF', action='store', \
                             help='CDF parameter of k mixture model prune', \
-                            default='0.99')
+                            default=0.99, type=float)
 
     parser.add_argument('tryname', action='store', \
                             help='the storage directory')