MCPcopy Create free account
hub / github.com/cjrd/self-supervised-pretraining / parse_args

Function parse_args

utils/plot_augmentation_modified.py:15–31  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

13dir_path = os.path.dirname(os.path.realpath(__file__))
14
15def parse_args():
16 parser = argparse.ArgumentParser(description='')
17 parser.add_argument(
18 '--results-dir',
19 default=os.path.realpath(os.path.join(dir_path, "results_modified")))
20 parser.add_argument(
21 '--out-dir',
22 default=os.path.realpath(os.path.join(dir_path, "plot-results")))
23 parser.add_argument(
24 '--dataset',
25 default="all")
26 parser.add_argument(
27 '--basetrain',
28 default="")
29
30 args = parser.parse_args()
31 return args
32
33#only deals with linear plots
34def gen_plots(data, options):

Callers 1

Calls

no outgoing calls

Tested by

no test coverage detected