From 114342747eaf4e5719745c0cd113272e3c4c51ba Mon Sep 17 00:00:00 2001
From: Peng Wu <alexepico@gmail.com>
Date: Mon, 25 Jul 2011 11:10:36 +0800
Subject: refine arguments

---
 tryprune.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'tryprune.py')

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')
-- 
cgit