From 7c08d87d63362b2acd2b6e8358b83f6b9b9a5c6b Mon Sep 17 00:00:00 2001 From: Benjamin Bolte Date: Thu, 1 Sep 2016 02:07:33 +0000 Subject: [PATCH] fixed bug --- generate_insurance_qa_embeddings.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/generate_insurance_qa_embeddings.py b/generate_insurance_qa_embeddings.py index 68097c7..081041f 100755 --- a/generate_insurance_qa_embeddings.py +++ b/generate_insurance_qa_embeddings.py @@ -1,5 +1,10 @@ #!/usr/bin/env python +""" +Command-line script for generating embeddings +Useful if you want to generate larger embeddings for some models +""" + from __future__ import print_function import os @@ -27,8 +32,8 @@ except KeyError: # parse arguments parser = argparse.ArgumentParser(description='Generate embeddings for the InsuranceQA dataset') -parser.add_argument('--iter', metavar='N', type=int, nargs=1, default=10, help='number of times to run') -parser.add_argument('--size', metavar='D', type=int, nargs=1, default=100, help='dimensions in embedding') +parser.add_argument('--iter', metavar='N', type=int, default=10, help='number of times to run') +parser.add_argument('--size', metavar='D', type=int, default=100, help='dimensions in embedding') args = parser.parse_args() # configure logging