Skip to content

Commit

Permalink
Add help to command line args for holdouts
Browse files Browse the repository at this point in the history
  • Loading branch information
justaddcoffee committed Sep 23, 2020
1 parent 21f46a6 commit 618ea3e
Showing 1 changed file with 10 additions and 5 deletions.
15 changes: 10 additions & 5 deletions run.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 618ea3e

Please sign in to comment.