Skip to content

Instantly share code, notes, and snippets.

@oscar-defelice

oscar-defelice/estimator.py

Created March 17, 2021 15:37
Show Gist options
  • Save oscar-defelice/960865fbb5f0e4eac342346fe9abfea0 to your computer and use it in GitHub Desktop.
Save oscar-defelice/960865fbb5f0e4eac342346fe9abfea0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
estimator.py
# estimator.py
import click
from utils import read_config, train
@click.command()
@click.argument('operation')
@click.option('--conf', type=click.Path(exists=True))
def main(operation, conf):
click.echo(f"{operation} started")
if operation == 'train':
config = read_config(conf)
train(config)
if __name__ == "__main__":
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment