diff options
Diffstat (limited to 'gen_exps.py')
-rwxr-xr-x | gen_exps.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/gen_exps.py b/gen_exps.py index 6488cdc..b847661 100755 --- a/gen_exps.py +++ b/gen_exps.py | |||
@@ -7,6 +7,7 @@ import re | |||
7 | import shutil as sh | 7 | import shutil as sh |
8 | import sys | 8 | import sys |
9 | 9 | ||
10 | from config.config import DEFAULTS | ||
10 | from optparse import OptionParser | 11 | from optparse import OptionParser |
11 | 12 | ||
12 | def parse_args(): | 13 | def parse_args(): |
@@ -15,7 +16,7 @@ def parse_args(): | |||
15 | 16 | ||
16 | parser.add_option('-o', '--out-dir', dest='out_dir', | 17 | parser.add_option('-o', '--out-dir', dest='out_dir', |
17 | help='directory for data output', | 18 | help='directory for data output', |
18 | default=("%s/exps"%os.getcwd())) | 19 | default=("%s/%s"% (os.getcwd(), DEFAULTS['out-gen']))) |
19 | parser.add_option('-f', '--force', action='store_true', default=False, | 20 | parser.add_option('-f', '--force', action='store_true', default=False, |
20 | dest='force', help='overwrite existing data') | 21 | dest='force', help='overwrite existing data') |
21 | parser.add_option('-n', '--num-trials', default=1, type='int', dest='trials', | 22 | parser.add_option('-n', '--num-trials', default=1, type='int', dest='trials', |
@@ -51,9 +52,9 @@ def main(): | |||
51 | if opts.described != None: | 52 | if opts.described != None: |
52 | for generator in opts.described.split(','): | 53 | for generator in opts.described.split(','): |
53 | if generator not in gen.get_generators(): | 54 | if generator not in gen.get_generators(): |
54 | print("No generator '%s'" % generator) | 55 | sys.stderr.write("No generator '%s'\n" % generator) |
55 | else: | 56 | else: |
56 | sys.stdout.write("Generator '%s', " % generator) | 57 | print("Generator '%s', " % generator) |
57 | gen.get_generators()[generator]().print_help() | 58 | gen.get_generators()[generator]().print_help() |
58 | if opts.list_gens or opts.described: | 59 | if opts.list_gens or opts.described: |
59 | return 0 | 60 | return 0 |
@@ -85,7 +86,7 @@ def main(): | |||
85 | if gen_name not in gen.get_generators(): | 86 | if gen_name not in gen.get_generators(): |
86 | raise ValueError("Invalid generator '%s'" % gen_name) | 87 | raise ValueError("Invalid generator '%s'" % gen_name) |
87 | 88 | ||
88 | print("Creating experiments using %s generator..." % gen_name) | 89 | sys.stderr.write("Creating experiments with %s generator...\n" % gen_name) |
89 | 90 | ||
90 | params = dict(gen_params.items() + global_params.items()) | 91 | params = dict(gen_params.items() + global_params.items()) |
91 | clazz = gen.get_generators()[gen_name] | 92 | clazz = gen.get_generators()[gen_name] |
@@ -94,5 +95,7 @@ def main(): | |||
94 | 95 | ||
95 | generator.create_exps(opts.out_dir, opts.force, opts.trials) | 96 | generator.create_exps(opts.out_dir, opts.force, opts.trials) |
96 | 97 | ||
98 | sys.stderr.write("Experiments saved in %s.\n" % opts.out_dir) | ||
99 | |||
97 | if __name__ == '__main__': | 100 | if __name__ == '__main__': |
98 | main() | 101 | main() |