From 618ea3e4b9427fa56770809b748fee4474753381 Mon Sep 17 00:00:00 2001 From: Justin Reese Date: Wed, 23 Sep 2020 16:42:19 -0700 Subject: [PATCH] Add help to command line args for holdouts --- run.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/run.py b/run.py index 8f5f6844..ec17bb67 100644 --- a/run.py +++ b/run.py @@ -112,11 +112,16 @@ def query(yaml: str, output_dir: str, @cli.command() -@click.option("nodes", "-n", default="data/merged/nodes.tsv", type=click.Path(exists=True)) -@click.option("edges", "-e", default="data/merged/edges.tsv", type=click.Path(exists=True)) -@click.option("output_dir", "-o", default="data/holdouts/", type=click.Path()) -@click.option("train_fraction", "-t", default=0.8, type=float) -@click.option("validation", "-v", is_flag=True, default=False) +@click.option("nodes", "-n", help="nodes KGX TSV file", default="data/merged/nodes.tsv", + type=click.Path(exists=True)) +@click.option("edges", "-e", help="edges KGX TSV file", default="data/merged/edges.tsv", + type=click.Path(exists=True)) +@click.option("output_dir", "-o", help="output directory", default="data/holdouts/", + type=click.Path()) +@click.option("train_fraction", "-t", + help="fraction of input graph to use in training graph [0.8]", + default=0.8, type=float) +@click.option("validation", "-v", help="make validation set", is_flag=True, default=False) def holdouts(*args, **kwargs) -> None: """Make holdouts for ML training