Esempio n. 1
0
def cli(ctx: click.Context, verbosity: str, batch_mb: float, nworkers: int) -> int:
    """Extract features and targets for training, testing and prediction."""
    ctx.obj = CliArgs(nworkers, batch_mb)
    configure_logging(verbosity)
    return 0
Esempio n. 2
0
def cli(ctx: click.Context, verbosity: str, batch_mb: float) -> int:
    """Train and predict using scikit-learn style models (in memory!)."""
    ctx.obj = CliArgs(batchMB=batch_mb)
    configure_logging(verbosity)
    return 0
Esempio n. 3
0
def cli(ctx: click.Context, gpu: bool, verbosity: str, batch_mb: float) -> int:
    """Train a model and use it to make predictions."""
    ctx.obj = CliArgs(gpu=gpu, batchMB=batch_mb)
    configure_logging(verbosity)
    return 0